python - Matplotlib：如何将散点图和线图组合到一个图例条目

Question

我使用 ax.scatter() 绘制数据点，并使用 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()

标签： pythonmatplotliblegend

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()