Pandas：如何将Pandas数据框中的两行变为列标题

在本文中，我们将介绍如何将Pandas数据框中的两行变为列标题。Pandas是一个基于NumPy的开源数据分析工具，为数据分析提供了高性能、易用的数据结构和数据分析工具。当处理数据时，经常需要将数据以特定格式进行展示，例如将数据框中的两行变为列标题，这个过程在Pandas中非常方便。

阅读更多：Pandas 教程

查看示例数据

下面我们先创建一个示例数据用于本文演示。

import pandas as pd

# 创建示例数据
data = {'Name': ['William', 'Emma', 'Sophia', 'Henry'], 
        "Gender": ['Male', 'Female', 'Female', 'Male'], 
        "Subject": ['Math', 'English', 'Math', 'English'], 
        "Score": [80, 90, 75, 85]}

df = pd.DataFrame(data)

# 查看示例数据
print(df)

运行上述代码，输出以下示例数据：

      Name  Gender  Subject  Score
0  William    Male     Math     80
1     Emma  Female  English     90
2   Sophia  Female     Math     75
3    Henry    Male  English     85

将两行变为列标题

假设我们想将“Gender”和“Subject”这两行变为列标题。可以使用Pandas的pivot()方法来实现这个过程。

# 将数据框的行索引设置为“Name”列
df = df.set_index('Name')

# 使用pivot()方法将“Gender”和“Subject”这两行变为列标题
df = df.pivot(columns='Gender', values='Score')
df.columns = df.columns.str.lower() + '_' + df.columns.name.str.lower()
df = df.join(pd.DataFrame(df.pop('male_score').values.tolist(), 
                          columns=['male_' + c for c in df.columns], 
                          index=df.index))
df = df.join(pd.DataFrame(df.pop('female_score').values.tolist(), 
                          columns=['female_' + c for c in df.columns], 
                          index=df.index))
df = df.reset_index()

# 查看转换后的数据
print(df)

上述代码中，我们首先将数据框的行索引设置为“Name”列，然后使用pivot()方法将“Gender”和“Subject”这两行变为列标题。接着，我们对列名进行修改，并使用join()方法将“male_score”和“female_score”两列合并到数据框中。最后，我们重新设置行索引为默认值，得到转换后的数据。

运行上述代码，输出以下转换后的数据：

      Name  female_math_score  female_english_score  male_math_score  male_english_score
0     Emma                NaN                  90.0              NaN                 85.0
1    Henry                NaN                  85.0              NaN                 80.0
2   Sophia               75.0                   NaN              NaN                  NaN
3  William               80.0                   NaN              NaN                  NaN