pandas列展开为行|极客教程

pandas列展开为行

在数据处理和分析中，经常会遇到需要将列展开为行的情况，这时候就需要使用pandas来实现这一功能。pandas是一个强大的数据处理库，在Python中被广泛应用于数据清洗、处理和分析中。本文将详细介绍如何使用pandas将列展开为行。

首先，我们先创建一个示例数据集，包含两列数据：姓名和爱好。我们将展示如何将这两列数据展开为多行数据。

import pandas as pd

data = {
    '姓名': ['张三', '李四'],
    '爱好': ['篮球,足球', '羽毛球,乒乓球']
}

df = pd.DataFrame(data)
print(df)

运行上面的代码，我们可以得到如下示例数据集：

   姓名       爱好
0  张三  篮球,足球
1  李四  羽毛球,乒乓球

接下来，我们将使用pandas的str.split()方法将列展开为行。具体步骤如下：

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

运行上面的代码，我们可以得到将列展开为行后的数据集：

   姓名  爱好
0  张三  篮球
1  张三  足球
2  李四  羽毛球
3  李四  乒乓球

通过以上代码，我们成功将原始数据集中的爱好列进行了展开，每个爱好都对应了一个新的行。

除了将单列展开为多行外，有时候我们还需要对多列进行展开。下面我们展示一个示例，将具有相同索引的多列同时展开为多行。