python - Pandas read_csv 摆脱双引号

Question

这是我的例子:

我首先创建数据框并将其保存到文件

import pandas as pd
df = pd.DataFrame({'col_1':[['a','b','s'], 23423]})
df.to_csv(r'C:\test.csv')

然后 df.col_1[0] 返回 ['a','b','s'] 一个列表

后来我从文件中读取了它:

df_1 = pd.read_csv(r'C:\test.csv', quoting = 3, quotechar = '"')

现在 df_1['col_1'][0] 返回 "['a' 's']" 一个字符串。

我想取回列表。我正在尝试不同的 read_csv 设置，但到目前为止运气不好

Answer 1

如果不做一些工作，您将无法取回列表

使用literal_eval转换列表

import ast

conv = dict(col_1=ast.literal_eval)
pd.read_csv(r'C:\test.csv', index_col=0, converters=conv).loc[0, 'col_1']

['a', 'b', 'c']