我的数据集中有一列由以下字符串列表组成:
列
['tablet','mobile','man']
[‘人’,‘房子’]
["人","计算机"]
["计算机"、"笔记本电脑"]
如何获取包含该列元素的列表?我想要这样的输出:
[[‘tablet’,‘mobile’,‘man’],[[‘people’,‘house’]。,["人"、"计算机"]、["计算机"、"笔记本电脑"]]
在列表理解中使用ast.literal_eval
转换字符串列表:
import ast
L = [ast.literal_eval(x) for x in df["column"]]
print (L)
[['tablet', 'mobile', 'man'], ['people', 'house'],
['people', 'computer'], ['computer', 'laptop']]
如果df
类似
column
0 ['tablet', 'mobile, man']
1 ['people', 'house']
2 ['people, 'computer']
3 ['computer', 'laptop']
您可以通过使用tolist()
调用列来获取列表
df["column"].tolist()
输出:
[['tablet', 'mobile', 'man'], ['people', 'house'], ['people', 'computer'], ['computer', 'laptop']]
import pandas as pd
result_list = []
df = pd.read_csv("file_name")
series_1 = df.loc[::, "column_name"]
for s in series_1:
list1 = s.strip("[]").replace(" ", "").replace("'", "").split(",")
result_list.append(s)
print(result_list)