是的,您可以使用pandas
和matplotlib
或seaborn
库按groupby
绘制多个柱状图。以下是一个示例:
首先,我们需要导入所需的库:
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
接下来,我们创建一个示例数据集:
data = {'Category': ['A', 'A', 'B', 'B', 'C', 'C'],
'Subcategory': ['X', 'Y', 'X', 'Y', 'X', 'Y'],
'Value': [10, 15, 20, 25, 30, 35]}
df = pd.DataFrame(data)
现在我们可以使用groupby
对数据进行分组,并绘制多个柱状图:
使用matplotlib
:
# 对数据进行分组
grouped = df.groupby(['Category', 'Subcategory'])['Value'].sum().unstack()
# 绘制柱状图
grouped.plot(kind='bar', figsize=(10, 5))
plt.title('Multiple Bar Charts by Group')
plt.xlabel('Category')
plt.ylabel('Value')
plt.xticks(rotation=0)
plt.legend(title='Subcategory')
plt.show()
使用seaborn
:
# 对数据进行分组
grouped = df.groupby(['Category', 'Subcategory'])['Value'].sum().reset_index()
# 绘制柱状图
sns.barplot(data=grouped, x='Category', y='Value', hue='Subcategory')
plt.title('Multiple Bar Charts by Group')
plt.xlabel('Category')
plt.ylabel('Value')
plt.legend(title='Sublabel')
plt.show()
这两个示例都将根据Category
和Subcategory
列对数据进行分组,并绘制多个柱状图。您可以根据需要调整代码以适应您的数据集和需求。
领取专属 10元无门槛券
手把手带您无忧上云