creating a matplotlib scatter legend size related

前端 未结 6 2065
野性不改
野性不改 2020-12-04 20:46

I am looking for a way to include a (matplotlib) legend that describe the size of points in a scatter plot, as this could be related to another variable, like in this basic

6条回答
  •  眼角桃花
    2020-12-04 21:08

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

提交回复
热议问题