Pandas实战100例 | 案例 59: 展开列表

案例 59: 展开列表

知识点讲解

当 DataFrame 中的某列包含列表或类似的数据结构时，Pandas 的 explode 方法可以用于将列表中的每个元素转换为单独的行。这对于处理嵌套数据特别有用。

• 展开列表: explode 方法将 DataFrame 的列中的列表元素展开，形成多行。每个列表元素都会成为新行，其他列的值被复制。

示例代码

# 准备数据和示例代码的运行结果，用于案例 59

# 示例数据
data_explode_lists = {
    'Category': ['A', 'B'],
    'Values': [[1, 2, 3], [4, 5]]
}
df_explode_lists = pd.DataFrame(data_explode_lists)

# 展开列表
exploded_df = df_explode_lists.explode('Values')

df_explode_lists, exploded_df

在这个示例中，我们使用 explode 方法将 Values 列中的列表展开为单独的行。

示例代码运行结果

原始 DataFrame (df_explode_lists):

  Category     Values
0        A  [1, 2, 3]
1        B     [4, 5]

展开后的 DataFrame (exploded_df):

  Category Values
0        A      1
0        A      2
0        A      3
1        B      4
1        B      5

这个结果展示了如何将嵌套列表中的元素展开成独立的行。这种操作在处理嵌套数据结构时非常有用。