假设我们有一个数据框df,其中包含“日期”、“城市”和“销售额”三列。现在我们想要旋转“销售额”列,并将其变成数据框的列名。但是我们希望保留其他两列。
我们可以使用melt()和pivot()函数来完成这个操作。具体流程如下:
# 导入所需的库
import pandas as pd
# 定义一个示例数据框
df = pd.DataFrame({'日期': ['2019-01-01', '2019-01-02', '2019-01-03', '2019-01-01', '2019-01-02', '2019-01-03'],
'城市': ['北京', '北京', '北京', '上海', '上海', '上海'],
'销售额': [100, 120, 80, 80, 90, 95]})
# 使用melt()函数将“销售额”列变成行,保留其他两列
df_melted = pd.melt(df, id_vars=['日期', '城市'], value_vars=['销售额'], var_name='指标', value_name='值')
# 使用pivot()函数将“销售额”列旋转回来
df_pivoted = df_melted.pivot(index=['日期', '城市'], columns='指标', values='值')
# 重置列名
df_pivoted.reset_index(inplace=True)
# 输出结果
print(df_pivoted)
输出结果如下:
指标 日期 城市 销售额
0 2019-01-01 上海 80
1 2019-01-01 北京 100
2 2019-01-02 上海 90
3 2019-01-02 北京 120
4 2019-01-03