Pandas DataFrame将列扩展为多行

在本文中，我们将介绍使用Python Pandas库将DataFrame中的列扩展为多行的方法。在数据分析中，扩展列可以使数据更可读和更方便分析。下面将详细讲解如何将单个列的值拆分为多行，并将其与其他列相匹配。

导入必要的库

在开始扩展列之前，我们需要导入必要的库：Pandas和NumPy。Pandas用于数据操作，NumPy用于科学计算。以下是导入这些库的代码：

import pandas as pd
import numpy as np

创建示例DataFrame

为了演示如何将列扩展为多行，我们将创建一个示例DataFrame。该DataFrame包含规格（Specification）、数量（Quantity）和颜色（Color）三列。

data = {'Specification': ['A5', 'A4', 'A3'], 
        'Quantity': [3, 2, 5], 
        'Color': ['红色, 蓝色, 黄色', '绿色, 黄色', '红色, 蓝色, 黑色, 白色, 灰色']}
df = pd.DataFrame(data)
print(df)

输出结果为：

  Specification  Quantity                        Color
0            A5         3                红色, 蓝色, 黄色
1            A4         2                      绿色, 黄色
2            A3         5  红色, 蓝色, 黑色, 白色, 灰色

将列扩展为多行

现在我们将使用Pandas的str.split方法，将Color列的元素（即每行中的字符串）拆分为多个元素，并将它们与其他列中的数据一起提取。以下是相应的代码：

df = df.assign(Color=df['Color'].str.split(', '))
df = df.explode('Color').reset_index(drop=True)
print(df)

输出结果为：

  Specification  Quantity Color
0            A5         3    红色
1            A5         3    蓝色
2            A5         3    黄色
3            A4         2    绿色
4            A4         2    黄色
5            A3         5    红色
6            A3         5    蓝色
7            A3         5    黑色
8            A3         5    白色
9            A3         5    灰色

我们可以看到，Color列被拆分成了多个行，并且与Specification和Quantity列中的数据匹配。在这里，我们使用了Pandas的assign方法来替换原始的Color列，并使用explode方法将字符列拆分成单个行。最后，我们使用reset_index方法重置索引值。

将多个字符列扩展为多行

在现实世界中，DataFrame中可能不止一个字符列需要扩展。在这种情况下，我们需要使用melt方法，可以在一次操作中将多个字符列扩展为多行。以下是相应的代码：

data = {'Specification': ['A5', 'A4', 'A3'], 
        'Quantity': [3, 2, 5], 
        'Color': ['红色, 蓝色, 黄色', '绿色, 黄色', '红色, 蓝色, 黑色, 白色, 灰色'],
        'Size': ['S, M', 'L', 'M, L, XL']}
df = pd.DataFrame(data)

df = df.melt(id_vars=['Specification', 'Quantity'], value_vars=['Color', 'Size'], var_name='Column')
df = df.assign(**{'ColumnValue': df.pop('value').str.split(', ')})
df = df.explode('ColumnValue').reset_index(drop=True)

print(df)

输出结果为：

  Specification  Quantity Column ColumnValue
0            A5         3  Color       红色
1            A5         3  Color       蓝色
2            A5         3  Color       黄色
3            A4         2  Color       绿色
4            A4         2  Color       黄色
5            A3         5  Color       红色
6            A3         5  Color       蓝色
7            A3         5  Color       黑色
8            A3         5  Color       白色
9            A3         5  Color       灰色
10           A5         3   Size          S
11           A5         3   Size          M
12           A4         2   Size          L
13           A3         5   Size          M
14           A3         5   Size          L
15           A3         5   Size         XL

在这里，我们首先使用melt方法将多个字符列转换为长格式，即将多列“融合”为一列。id_vars参数指定保留的列，value_vars参数指定要转换的列，var_name参数指定新列的名称。在这里，我们将两个属性列（Color和Size）转成Column列，值存放在ColumnValue中。

然后，我们使用assign方法来将新列转换为列表，并使用explode方法将它们拆分成单行。最后，我们使用reset_index方法重置索引值。

此外，我们也可以使用类似的方法将其他类型的列扩展成多行，如时间序列数据。这个方法依然是将一个列的值拆分成多行，仅仅是Split的方式需要根据数据的不同类型做相应的调整。