python - 为来自矩阵一列的每个线图创建图例条目

Question

我绘制了有 4 列的矩阵 nH.T。所以我想制作图例，每列都有列表 latent_prot2 中的名称。

latent_prot2 = np.array(["A","B","C","E"])

ax3 = plt.subplot2grid((3, 4), (2, 0), colspan=3, rowspan=1)
ax3.plot(nH.T, label=[n for i,n in enumerate(latent_prot2)])
plt.legend()

我得到的是每行打印出来的所有列表。

我尝试了 ax3.plot(nH.T, label=[n[i] for i,n in enumerate(latent_prot2)]) 或将最后一行更改为 plt.legend (label=latent_prot2) 但它们不起作用。如何在列表中的图例中写入矩阵中每一列数据的名称？如果我不必通过 for 循环，那就太好了，但我不知道这是否可能。

Answer 1

使用 plot 的 label 参数，您将为每条绘制的曲线获得相同的标签。

要操作图例，您可以将元素列表传递给 legend 调用

ax.legend(labels=["A","B","C","E"])

完整示例:

import matplotlib.pyplot as plt
import numpy as np

a = np.cumsum(np.cumsum(np.random.randn(15,4), axis=0), axis=1)

lab = np.array(["A","B","C","E"])

fig, ax = plt.subplots()
ax.plot(a)
ax.legend(labels=lab )

plt.show()