python - 设置图例的最大行数

Question

我需要在一个图中绘制多个数据集。数据集的数量各不相同，所以我事先不知道会有多少。

如果我只是绘制图例，我会得到这个(下面的 MCVE):

我如何告诉 plt.legend() 只绘制前 10 个图例？我环顾了 plt.legends()类，但似乎没有理由设置这样的值。

MCVE:

import numpy as np
import matplotlib.pyplot as plt

dataset = []
for _ in range(20):
    dataset.append(np.random.uniform(0, 1, 2))

lbl = ['adfg', 'dfgb', 'cgfg', 'rtbd', 'etryt', 'frty', 'jklg', 'jklh',
       'ijkl', 'dfgj', 'kbnm', 'bnmbl', 'qweqw', 'fghfn', 'dfg', 'hjt', 'dfb',
       'sdgdas', 'werwe', 'dghfg']

for i, xy in enumerate(dataset):
    plt.scatter(xy[0], xy[1], label=lbl[i])
plt.legend()
plt.savefig('test.png')

Answer 1

您可以限制显示的标签数量。

import matplotlib.pyplot as plt

maxn = 16
for i in range(25):
    plt.scatter(.5, .5, label=(i//maxn)*"_"+str(i))
plt.legend()
plt.show()

这个方法当然也适用于文本标签:

import numpy as np
import matplotlib.pyplot as plt

labels = ["".join(np.random.choice(list("ABCDEFGHIJK"), size=8)) for k in range(25)]
maxn = 16
for i,l in enumerate(labels):
    plt.scatter(.5, .5, label=(i//maxn)*"_"+l)
plt.legend()
plt.show()

这样做的原因是图例中忽略了以 "_" 开头的标签。这在内部用于为对象提供标签而不在图例中显示它们，但我们当然也可以使用它来限制图例中的元素数量。