python – 如何“解包”包含多个项目的列？

2024年1月7日 205次阅读

我有一个DataFrame,其中一列是一个集合的字符串表示.是否有可能“解开”任何这样的行？

一个例子：

     ColA                 ColB                    ColC   ColD
0    "one item in a set"  "{'item 1'}"            "..."  "..."
1    "several in a set"   "{'item 1', 'item 2'}"  "..."  "..."
...

应该成为：

     ColA                 ColB       ColC   ColD 
0    "one item in a set"  'item 1'   "..."  "..."
1    "several in a set"   'item 1'   "..."  "..." 
2    "several in a set"   'item 2'   "..."  "..."
...

最佳答案 你可以：

df2 = df.colB.str[1:-1].str.split(',', expand=True)
df2 = df2.stack().reset_index()
df2 = df2.drop('level_1', axis=1).rename(columns={0: 'colB'}).set_index('level_0')
df = df.drop('colB', axis=1)
df = pd.concat([df, df2], axis=1)

删除{}后,.split()on’,’展开到新列和.stack(),然后清理.

点赞