首页 > 后端开发 > Python教程 > 如何将 Pandas DataFrame 中的逗号分隔值拆分为单独的行?

如何将 Pandas DataFrame 中的逗号分隔值拆分为单独的行?

Mary-Kate Olsen
发布: 2024-12-24 18:06:15
原创
268 人浏览过

How to Split Comma-Separated Values in a Pandas DataFrame into Separate Rows?

将 Pandas 数据框字符串条目拆分(分解)到单独的行

问题:
操作 Pandas 数据框包含一列逗号分隔值,目标是拆分每个值将 CSV 字段分成单独的行,保留原始数据结构。

解决方案:

推荐的解决方案是利用 Pandas Series.explode() 或 DataFrame.explode( ) 方法,在 Pandas 0.25.0 中引入,并在 Pandas 1.3.0 中增强以支持多列爆炸。

要爆炸单个列,请使用 Series.explode():

df.explode('column_name')
登录后复制

对于多列,请使用DataFrame.explode():

df.explode(['column1', 'column2'])
登录后复制

示例:

df = pd.DataFrame({
    'A': [[0, 1, 2], 'foo', [], [3, 4]],
    'B': 1,
    'C': [['a', 'b', 'c'], np.nan, [], ['d', 'e']]
})

df.explode('A')
登录后复制

输出:

     A  B          C
0    0  1  [a, b, c]
0    1  1  [a, b, c]
0    2  1  [a, b, c]
1  foo  1        NaN
2  NaN  1         []
3    3  1     [d, e]
3    4  1     [d, e]
登录后复制

适用于多个普通列和列表列的更通用方法,考虑以下函数:

def explode(df, lst_cols, fill_value='', preserve_index=False):
    # Ensure `lst_cols` is list-alike
    if lst_cols and not isinstance(lst_cols, (list, tuple, np.ndarray, pd.Series)):
        lst_cols = [lst_cols]

    # Calculate lengths of lists
    lens = df[lst_cols[0]].str.len()

    # Preserve original index values
    idx = np.repeat(df.index.values, lens)

    # Create an "exploded" DataFrame
    res = (pd.DataFrame({
                col:np.repeat(df[col].values, lens)
                for col in df.columns.difference(lst_cols)
            },
            index=idx)
             .assign(**{col:np.concatenate(df.loc[lens>0, col].values)
                            for col in lst_cols}))

    # Append rows with empty lists
    if (lens == 0).any():
        res = (res.append(df.loc[lens==0, df.columns.difference(lst_cols)], sort=False)
                  .fillna(fill_value))

    # Revert to original index order and reset if requested
    res = res.sort_index()
    if not preserve_index:        
        res = res.reset_index(drop=True)

    return res
登录后复制

爆炸 a 的示例类似 CSV 的列:

df = pd.DataFrame({
    'var1': 'a,b,c d,e,f,x,y'.split(),
    'var2': [1, 2]
})

explode(df.assign(var1=df.var1.str.split(',')), 'var1')
登录后复制

输出:

  var1  var2
0    a     1
1    b     1
2    c     1
3    d     2
4    e     2
5    f     2
6    x     2
7    y     2
登录后复制

以上是如何将 Pandas DataFrame 中的逗号分隔值拆分为单独的行?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板