下面是一个按照单词数量进行堆叠条形图的代码示例:
import matplotlib.pyplot as plt
# 定义数据
labels = ['A', 'B', 'C', 'D']
word_counts1 = [10, 15, 7, 12]
word_counts2 = [8, 12, 10, 5]
word_counts3 = [5, 10, 15, 8]
# 创建堆叠条形图
fig, ax = plt.subplots()
# 绘制第一个数据集
ax.bar(labels, word_counts1, label='Data 1')
# 绘制第二个数据集
ax.bar(labels, word_counts2, bottom=word_counts1, label='Data 2')
# 绘制第三个数据集
ax.bar(labels, word_counts3, bottom=[i+j for i,j in zip(word_counts1, word_counts2)], label='Data 3')
# 添加图例
ax.legend()
# 添加标题和标签
ax.set_xlabel('Categories')
ax.set_ylabel('Word Counts')
ax.set_title('Word Counts by Category')
# 显示图形
plt.show()
在这个示例中,我们使用了matplotlib.pyplot
库来创建堆叠条形图。首先,我们定义了要显示的数据,包括标签和每个标签对应的单词数量。然后,我们使用ax.bar()
函数绘制了每个数据集的条形图,并使用bottom
参数指定了上方数据集的堆叠位置。最后,我们添加了图例、标题和标签,并使用plt.show()
函数显示图形。