【Pandas】pandas wide_to_long

Pandas2.2 General

Data manipulations

pandas.wide_to_long()

pandas.wide_to_long 是 Pandas 库中用于将宽格式数据转换为长格式数据的一个非常有用的函数。以下是该函数每个参数的详细介绍，以及一个示例和结果。

def wide_to_long(
    df: DataFrame, stubnames, i, j, sep: str = "", suffix: str = r"\d+"
) -> DataFrame:
    ...

参数详解

• df：

  • 类型：Pandas DataFrame。
  • 说明：包含要转换的宽格式数据的 DataFrame。

• stubnames：

  • 类型：字符串或字符串列表。
  • 说明：指定要转换的列名的前缀。这些前缀列名后面通常会跟着一些后缀（如数字或字母），用于区分不同的变量。wide_to_long 会查找这些前缀相同的列，并将它们转换为长格式数据中的不同观测值。

• i：

  • 类型：字符串或字符串列表。
  • 说明：指定用作标识符的列名。这些列的值在转换后的长格式数据中会保持不变，用于唯一标识每个观测值。

• j：

  • 类型：字符串。
  • 说明：指定新生成的列的名称。在转换后的长格式数据中，这个名称会用于表示原来的宽格式数据中不同后缀对应的变量。

• sep：

  • 类型：字符串。
  • 说明：指定用于分隔 stubnames 和 j 的连接符。在转换过程中，这个连接符会从列名中去除。默认为空字符串。

• suffix：

  • 类型：字符串。
  • 说明：指定用于捕获后缀的正则表达式。在宽格式数据中，这个正则表达式会用于匹配 stubnames 后面的后缀。默认为 '\d+'，表示一个或多个数字。

示例及结果

以下是一个使用 pandas.wide_to_long 的示例：

import pandas as pd

# 创建一个宽格式的 DataFrame
data = {
    'famid': [1, 1, 1, 2, 2, 2, 3, 3, 3],
    'birth': [1, 2, 3, 1, 2, 3, 1, 2, 3],
    'ht1': [2.8, 2.9, 2.2, 2, 1.8, 1.9, 2.2, 2.3, 2.1],
    'ht2': [3.4, 3.8, 2.9, 3.2, 2.8, 2.4, 3.3, 3.4, 2.9]
}
df = pd.DataFrame(data)

# 使用 wide_to_long 函数将宽格式数据转换为长格式数据
l = pd.wide_to_long(df, stubnames=['ht'], i=['famid', 'birth'], j='age')

# 打印转换后的长格式数据
print(l)

输出结果：

famid birth age
1     1     1    2.8
            2    3.4
      2     1    2.9
            2    3.8
      3     1    2.2
            2    2.9
2     1     1    2.0
            2    3.2
      2     1    1.8
            2    2.8
      3     1    1.9
            2    2.4
3     1     1    2.2
            2    3.3
      2     1    2.3
            2    3.4
      3     1    2.1
            2    2.9