python - 如何从列表创建分组条形图

Question

• 我正在尝试绘制条形图，该图比较 2 种不同情况下的多个项目的数量。
• 所需的输出是一个包含 4+4 = 8 个条形图的条形图，条形图彼此相邻，指示每种情况的每种类型的数量。
• 这是我编写的初始代码，它没有给出我所期望的。我该如何修改这个？

import numpy
import matplotlib.pyplot as plt

names = ["a","b","c","d"]
case1 = [5,7,5,6]
case2 = [7,4,8,5]

plt.hist(case1)
plt.show()

Answer 1

• 最简单的方法是使用 pandas 创建数据框，然后使用 pandas.DataFrame.plot 进行绘图
  • 数据帧索引(本例中为'names')自动用于 x 轴，并且各列绘制为条形。
  • matplotlib 用作绘图后端
• 在 python 3.8、pandas 1.3.1 和 matplotlib 3.4.2 中进行测试
• 有关长度不均匀的列表，请参阅 How to create a grouped bar plot from lists of uneven length

import pandas as pd
import matplotlib.pyplot as plt

names = ["a","b","c","d"]
case1 = [5,7,5,6]
case2 = [7,4,8,5]

# create the dataframe
df = pd.DataFrame({'c1': case1, 'c2': case2}, index=names)

# display(df)
   c1  c2
a   5   7
b   7   4
c   5   8
d   6   5

# plot
ax = df.plot(kind='bar', figsize=(6, 4), rot=0, title='Case Comparison', ylabel='Values')
plt.show()

• 针对 python 2.7 尝试以下操作

fig, ax = plt.subplots(figsize=(6, 4))
df.plot.bar(ax=ax, rot=0)
ax.set(ylabel='Values')
plt.show()