python - 如何将列表的数据框列折叠成一个 NLP 列表

Question

我正在尝试进行一些主题分析，我需要将数据框中由列表组成的列折叠到一个单词列表中。

所以这是我的数据的近似值：

import pandas as pd
d = {'Case': ["[wait, information, employer]","[case, assign, priority, level, 2, transmit]" ]}
df2 = pd.DataFrame(data=d)

我想得到一个列表，如 list = ['wait', 'information', 'case','assign','priority', 'level']

标签： pythonpandaslistdataframe

Answer 1

df2["CaseList"] = df2["Case"].apply(lambda x: x.replace("]","").replace("[","").split(","))

那对你有用吗？