python - 如何使用 Pandas 将字符串转换回列表

Question

我有一个包含一些数据的 txt 文件，其中一列是这样的:

['BONGO', 'TOZZO', 'FALLO', 'PINCO']

为了加载文件，我使用了 pandas 函数 to_csv。

加载数据框后，内容看起来没问题，但后来我意识到数据框中的项目不是项目列表，而是一个字符串，其元素是列表中的字符!

df['column'] 返回这样的字符串

"['BONGO', 'TOZZO', 'FALLO', 'PINCO']"

而不是像这样的列表:

['BONGO', 'TOZZO', 'FALLO', 'PINCO']

因此，如果我输入 df['column'][0]，我会得到 '[' 而不是 BONGO

我应该怎么做才能将字符串转换回其原始列表格式？我应该使用 to_csv 函数的任何输入吗？

Answer 1

您可以将 ast.literal_eval 用作:

>>> import ast
>>> a = "['BONGO', 'TOZZO', 'FALLO', 'PINCO']"
>>> print ast.literal_eval(a)
>>> ['BONGO', 'TOZZO', 'FALLO', 'PINCO']