时间戳

首页 > 解决方案 > 将字符串作为列表重塑为行

python - 将字符串作为列表重塑为行

问题描述

我有一个这样的熊猫数据框：

df = pandas.DataFrame({
        'Grouping': ["A", "B", "C"], 
        'Elements': ['[\"A1\"]', '[\"B1\", \"B2\", \"B3\"]', '[\"C1\", \"C2\"]']
    }).set_index('Grouping')

所以

            Elements
Grouping
===============================
A           ["A1"]
B           ["B1", "B2", "B3"]
C           ["C1", "C2"]

即一些列表被编码为字符串作为列表。什么是一种干净的方法可以将其重塑为一个整洁的数据集，如下所示：

            Elements
Grouping
====================
A           A1
B           B1
B           B2
B           B3
C           C1
C           C2

不诉诸for循环？我能想到的最好的是：

df1 = pandas.DataFrame()
for index, row in df.iterrows():
    df_temp = pandas.DataFrame({'Elements': row['Elements'].replace("[\"", "").replace("\"]", "").split('\", \"')})
    df_temp['Grouping'] = index
    df1 = pandas.concat([df1, df_temp])
df1.set_index('Grouping', inplace=True)

但这很丑陋。

标签： pythonpandas

解决方案

您可以使用.str.extractall()：

df.Elements.str.extractall(r'"(.+?)"').reset_index(level="match", drop=True).rename({0:"Elements"}, axis=1)

结果：

         Elements
Grouping         
A              A1
B              B1
B              B2
B              B3
C              C1
C              C2

推荐阅读