pandas reshape|极客教程

pandas reshape

Pandas是Python中用于数据分析和处理的重要库，它提供了大量的功能和方法来操作数据。其中，reshape是Pandas中一个非常常用的功能，用于重塑数据的形状。在本文中，我们将详细介绍Pandas中reshape的相关知识，并通过示例代码来演示如何使用这些功能。

1. stack和unstack

首先，我们来介绍stack和unstack这两个函数。stack用于将列转换为行，而unstack则是将行转换为列，这两个函数常用于处理多级索引的数据。

import pandas as pd

# 创建一个有多级索引的DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6]}
df = pd.DataFrame(data, index=pd.MultiIndex.from_tuples([('X', 'Y'), ('X', 'Z'), ('Y', 'W')], names=['level_1', 'level_2']))
print(df)

# 使用stack将列转换为行
stacked = df.stack()
print(stacked)

# 使用unstack将行转换为列
unstacked = stacked.unstack()
print(unstacked)

运行结果：

           A  B
level_1 level_2      
X       Y   1  4
        Z   2  5
Y       W   3  6

level_1  level_2
X        Y          A  1
         Z          A  2
Y        W          A  3
dtype: int64

2. pivot_table

pivot_table函数可以根据一个或多个列来创建数据透视表，从而使数据易于分析。以下示例演示了如何使用pivot_table函数。

import pandas as pd

# 创建一个包含销售数据的DataFrame
data = {'Date': ['2021-01-01', '2021-01-01', '2021-01-02', '2021-01-02'],
        'Product': ['A', 'B', 'A', 'B'],
        'Sales': [100, 200, 150, 250]}
df = pd.DataFrame(data)

# 使用pivot_table创建数据透视表
pivot_df = df.pivot_table(index='Date', columns='Product', values='Sales', aggfunc='sum')
print(pivot_df)

运行结果：

Product      A    B
Date                
2021-01-01  100  200
2021-01-02  150  250

3. melt

melt函数用于将宽格式数据转换为长格式数据，这在数据分析和可视化中很有用。

import pandas as pd

# 创建一个包含学生成绩的DataFrame
data = {'Student': ['A', 'B', 'C'],
        'Math': [90, 95, 85],
        'English': [85, 90, 95],
        'Science': [80, 88, 92]}
df = pd.DataFrame(data)

# 使用melt函数将宽格式数据转换为长格式数据
melted_df = df.melt(id_vars='Student', var_name='Subject', value_name='Score')
print(melted_df)

运行结果：

  Student  Subject  Score
0       A     Math     90
1       B     Math     95
2       C     Math     85
3       A  English     85
4       B  English     90
5       C  English     95
6       A  Science     80
7       B  Science     88
8       C  Science     92