从 Pandas DataFrame 绘制条形图

Question

假设我有一个看起来像这样的DataFrame：

Hour | V1 | V2 | A1 | A2
 0   | 15 | 13 | 25 | 37  
 1   | 26 | 52 | 21 | 45 
 2   | 18 | 45 | 45 | 25 
 3   | 65 | 38 | 98 | 14

我正在尝试创建一个条形图来比较 Hour 的列 V1 和 V2。 当我这样做时：

import matplotlib.pyplot as plt
ax = df.plot(kind='bar', title ="V comp",figsize=(15,10),legend=True, fontsize=12)
ax.set_xlabel("Hour",fontsize=12)
ax.set_ylabel("V",fontsize=12)

我得到一个带有所有列的值和名称的图和图例。如何修改我的代码，以便绘图和图例仅显示列 V1 和 V2

Answer 1

要仅绘制所选列，您可以通过将列表传递给下标运算符来选择感兴趣的列：

ax = df[['V1','V2']].plot(kind='bar', title ="V comp", figsize=(15, 10), legend=True, fontsize=12)

您尝试的是df['V1','V2']，这将引发KeyError，因为该标签不存在正确的列，尽管起初看起来很有趣，但您必须考虑到您正在传递一个列表，因此双方括号[[]] .

import matplotlib.pyplot as plt
ax = df[['V1','V2']].plot(kind='bar', title ="V comp", figsize=(15, 10), legend=True, fontsize=12)
ax.set_xlabel("Hour", fontsize=12)
ax.set_ylabel("V", fontsize=12)
plt.show()