可视化pandas分组数据

import pandas as pd

data = {
    'Category': ['A', 'B', 'A', 'B', 'A', 'B', 'A', 'B'],
    'Sales': [100, 150, 120, 170, 140, 160, 130, 180],
    'Region': ['North', 'North', 'South', 'South', 'East', 'East', 'West', 'West']
}

df = pd.DataFrame(data)

grouped_data = df.groupby(['Category', 'Region'])['Sales'].sum().reset_index()

import matplotlib.pyplot as plt

# 设置图形大小
plt.figure(figsize=(10, 6))

# 绘制条形图
for category in grouped_data['Category'].unique():
    category_data = grouped_data[grouped_data['Category'] == category]
    plt.bar(category_data['Region'], category_data['Sales'], label=category)

# 添加标题和标签
plt.title('Sales by Category and Region')
plt.xlabel('Region')
plt.ylabel('Sales')
plt.legend()

# 显示图形
plt.show()

import seaborn as sns

# 设置图形大小
plt.figure(figsize=(10, 6))

# 绘制折线图
sns.lineplot(data=grouped_data, x='Region', y='Sales', hue='Category')

# 添加标题和标签
plt.title('Sales by Category and Region')
plt.xlabel('Region')
plt.ylabel('Sales')

# 显示图形
plt.show()