Python学习笔记：一个stack、unstack真实案例

SHOW MY CODE!! NO BB!!

# -*- coding: utf-8 -*-
"""
Created on Fri Sep 17 16:32:54 2021

@author: Hider
"""

# 加载包
import pandas as pd
import os
os.chdir(r'C:\Users\Hider\Desktop')

# 测试数据
# df_test = pd.read_clipboard()

# 读取数据
data = pd.read_excel('data.xlsx', sheet_name='Sheet3')

# 第一版
df = data.set_index(['地市', '细分市场']).stack(level=0).reset_index().set_index(['地市', '细分市场', 'level_2']).unstack(level=1).reset_index()

# 优化
df = data.set_index(['地市', '细分市场']).stack().unstack(level=1).reset_index()

# 修改列名
df.columns = ['地市','月份','公众市场','农村市场','商客市场','城市家庭','政企市场','校园市场','流动市场','行客市场']

# 修改类型
df['月份'] = df['月份'].astype(int)

# 按月份排序
df2 = df.sort_values(by=['月份'])

# 输出顺序设置
output_columns = ['地市','月份','行客市场','公众市场','农村市场','商客市场','城市家庭','政企市场','校园市场','流动市场']

# 导出文件
df2.to_csv("result.csv", encoding="gbk", 
          index=False, columns=output_columns)