使用动态列数创建matplotlib图例

Question

我希望在matplotlib中创建一个图例，每列最多有5个条目。现在，我可以手动设置列数，例如：

leg = plt.legend(loc='best', fancybox=None, ncol=2)

如何修改此代码，以便每列最多只允许5个条目？

- user308827

Answer 1

没有内置的方法可以指定行数而不是列数。但是，您可以使用ax._get_legend_handles()方法获取将添加到图例中的项数。

例如：

import numpy as np
import matplotlib.pyplot as plt

numlines = np.random.randint(1, 15)
x = np.linspace(0, 1, 10)

fig, ax = plt.subplots()
for i in range(1, numlines + 1):
    ax.plot(x, i * x, label='$y={}x$'.format(i))

numitems = len(list(ax._get_legend_handles()))
nrows = 5
ncols = int(np.ceil(numitems / float(nrows)))

ax.legend(ncol=ncols, loc='best')

plt.show()

- Joe Kington