如何在matplotlib事件图中添加图例？

Posted 2021-03-31

有谁知道如何在matplotlib的新绘图类型eventplot中添加图例？

我没有成功尝试过很多东西，包括：

labels1 = ['red', 'green', 'blue', 'yellow', 'magenta', 'cyan']
ax2.eventplot(data1, colors=colors1, lineoffsets=lineoffsets1,
              linelengths=linelengths1, orientation='vertical', label=labels1)
ax2.legend()

这些编辑是在http://matplotlib.org/dev/api/pyplot_api.html#matplotlib.pyplot.eventplot上找到的示例代码eventplot_demo.py上进行的

答案

TL;DR

只是，ax.legend( list_of_names )

Explain

从你的代码，

labels1 = ['red', 'green', 'blue', 'yellow', 'magenta', 'cyan']
ax2.eventplot(data1, colors=colors1, lineoffsets=lineoffsets1,
              linelengths=linelengths1, orientation='vertical')
ax2.legend( labels1 )

以matplotlib.pyplot.eventplot为例

import matplotlib.pyplot as plt
import numpy as np
import matplotlib

# Fixing random state for reproducibility
np.random.seed(19680801)

# create random data
data = np.random.random([6, 50])

# set different colors for each set of positions
colors = np.array([[1, 0, 0],
                    [0, 1, 0],
                    [0, 0, 1],
                    [0, 0, 0],
                    [1, 0, 1],
                    [0, 1, 1]])
# label each set
labels = ['John','Mike','Bob','Sam','Alex','Eva']

fig, axs = plt.subplots(1,1)

# create a horizontal plot
axs.eventplot(data,color=colors)
# Just needed axs.legend(labels) the everything else is for the positioning
axs.legend(labels, bbox_to_anchor=(0., 1.0, 1., .10), loc=3,ncol=3, mode="expand", borderaxespad=0.)
plt.show()

最后结果