不完整的散点图图例：当前点的大小不够

Question

我想在图例中包含情节中出现的所有尺寸。我不知道为什么我只打印 1 种尺寸和 1 种价格。这是代码：

import numpy as np
import matplotlib.pyplot as plt

# canvas , blackboard learn, moodle, D2L, Edmodo
# green  ,    blue         ,yellow , red, orange

# x= CVSS v2 score (variabile 1)
x=[0.22,-1.4,-0.86,0.65,0.84]

# y= supported OS (variabile 2)
y=[1.16,-0.77,-1.25,0.68,0.19]

colors=['green','blue','black','red','orange']
labels=['Canvas','BlackBoard Learn','Moodle','D2L', 'Edmodo']
scale=[1.76,24,1.7,1.83,8.75]  

fig, ax = plt.subplots(figsize=(7,5))

for i in range(0,5):
    scale[i]*=200
    scatter=ax.scatter(x[i], y[i], s=scale[i], c=colors[i], label=labels[i], alpha=0.45)

    
lgnd= plt.legend(loc="upper right")
plt.xlabel('CVSS v2 score')
plt.ylabel('supported OS')
plt.ylim((-2,2))
plt.xlim((-1.5,4))

lgnd.legendHandles[0]._sizes = [60]
lgnd.legendHandles[1]._sizes = [60]
lgnd.legendHandles[2]._sizes = [60]
lgnd.legendHandles[3]._sizes = [60]
lgnd.legendHandles[4]._sizes = [60]
ax.add_artist(lgnd)

kw = dict(prop="sizes", num=20, color=scatter.cmap(0.1), fmt="$ {x:.2f}"+" / month", func=lambda s: s/200)
legend2 = ax.legend(*scatter.legend_elements(**kw), prop={'size':14}, loc="lower right", title="Price") 
plt.show()

这是输出：

我想要这样的东西：

Answer 1

可以通过创建没有数据点的虚拟艺术家作为图例的自定义句柄来实现，您可以在其中指定图例的外观。

此外，如果您在 matplotlib 中使用面向对象的方法，您应该保持一致，并使用面向对象的方法通过调用轴的相应方法来设置限制和轴标签（如果在代码中这样做你）。

import numpy as np
import matplotlib.pyplot as plt

# x= CVSS v2 score (variabile 1)
x=[0.22,-1.4,-0.86,0.65,0.84]

# y= supported OS (variabile 2)
y=[1.16,-0.77,-1.25,0.68,0.19]

colors = ['green','blue','black','red','orange']
labels = ['Canvas','BlackBoard Learn','Moodle','D2L', 'Edmodo']
prices = [1.76,24,1.7,1.83,8.75]
scales = [price * 200 for price in prices]

fig, ax = plt.subplots(figsize=(7,5))

ax.set_xlabel('CVSS v2 score')
ax.set_ylabel('supported OS')
ax.set_ylim((-2,2))
ax.set_xlim((-1.5,4))

for i in range(0,5):
    scatter=ax.scatter(x[i], y[i], s=scales[i], c=colors[i], label=labels[i], alpha=0.45)

labels1 = labels
handles1 = [ax.scatter([], [], s=60, c=color, alpha=0.45) for color in colors]
legend1 = ax.legend(handles=handles1, labels=labels1, loc="upper right")

labels2 = [f'$ {price} / month' for price in sorted(prices)]
handles2 = [ax.scatter([], [], s=size, c='purple', alpha=0.45) for size in sorted(scales)]
legend2 = ax.legend(handles=handles2, labels=labels2, loc="lower right")
ax.add_artist(legend1)

plt.show()

我的结果是这样的：

不完整的散点图图例：当前点的大小不够

incomplete scatter plot legend: not enough sizes for the present points

python

matplotlib

plot

legend

scatter