python - 创建一个 matplotlib 散点图例大小相关

Question

我正在寻找一种方法来包含一个 (matplotlib) 图例，该图例描述散点图中点的大小，因为这可能与另一个变量相关，就像在这个基本示例中一样:

import numpy as np
import matplotlib.pyplot as plt

N = 50
x = np.random.rand(N)
y = np.random.rand(N)
a2 = 400*np.random.rand(N)

plt.scatter(x, y, s=a2, alpha=0.5)
plt.show()

(灵感来自: http://matplotlib.org/examples/shapes_and_collections/scatter_demo.html )

所以在图例中，根据 scatter s 描述符，理想情况下对应于尺寸 0-400(a2 变量)的点很少.

Answer 1

使用.legend_elements("sizes") :

import numpy as np
import matplotlib.pyplot as plt

N = 50
x = np.random.rand(N)
y = np.random.rand(N)
a2 = 400*np.random.rand(N)

sc = plt.scatter(x, y, s=a2, alpha=0.5)
plt.legend(*sc.legend_elements("sizes", num=6))
plt.show()