matplotlib - 控制图例中的行数

Question

我目前正在尝试在单个绘图上绘制大量数据。我使用重复的颜色和符号来构建我的表示。但是，在绘制最终结果时，图例显得略有偏差，因为我无法控制其中的行数。因此，我得到的是 5 -4 -4 -4 (我更喜欢 5 - 5 - 5 - 2)，而不是得到 5 个重复的绿色，然后是 5 个重复的红色，5 个重复的蓝色，然后是 2 个其他。

您可以在附图中清楚地看到这一点。

现在我对图例使用这些选项:

axp.legend(loc="lower right",ncol=4)

Answer 1

我也遇到过这个问题几次，并通过向图例添加虚拟项来填充最后一列来使用此解决方法，如果有更优雅的方法可用，我也将非常有兴趣了解它们。

import numpy as np
import matplotlib.pylab as pl

pl.figure()

pl.plot(np.arange(10), np.random.random([10,5]), color='r', label='red')
pl.plot(np.arange(10), np.random.random([10,5]), color='g', label='green')
pl.plot(np.arange(10), np.random.random([10,5]), color='b', label='blue')
pl.plot(np.arange(10), np.random.random([10,2]), color='k', label='black')

# Add empty dummy legend items
pl.plot(np.zeros(1), np.zeros([1,3]), color='w', alpha=0, label=' ')

pl.legend(ncol=4)