Matplotlib:如何将散点图和折线图组合为一个图例条目

Question

我用ax.stratter((绘制数据点，并用ax.plot((连接拟合的数据点。

如何在图例中创建一个通用条目，将数据点的标记与拟合线相结合？我想获得一个图例条目，就像我为ax.plot(x, y, '-o', label = 'abc')获得它一样。

我创建了以下最小的示例：

import matplotlib.pyplot as plt
import numpy as np
x_scatter = np.linspace(0,10,10)
x_line = np.linspace(0,10,100)
fig, ax = plt.subplots()
for i in range(5):
ax.scatter(x_scatter, np.sin(x_scatter) + i, label = i)
ax.plot(x_line, np.sin(x_line)+i)
plt.legend(loc='best')
plt.show()

Answer 1

这个"破解"应该有效：

import matplotlib.pyplot as plt
import numpy as np
x_scatter = np.linspace(0,10,10)
x_line = np.linspace(0,10,100)
fig, ax = plt.subplots()
prop = ax._get_lines.prop_cycler
for i in range(5):
color = next(prop)['color']
ax.scatter(x_scatter, np.sin(x_scatter) + i, color=color)
ax.plot(x_line, np.sin(x_line)+i, color=color)
ax.plot([], [], '-o', color=color, label = i)
plt.legend(loc='best')
plt.show()