python - 将包含元素列表的 csv 文件读入 pandas 数据框

Question

我有一个 CSV 文件，格式如下:

index      A      B      C
ind1    [1,2,3][3,4,5][6,7,8]
ind2    [1,4,3,4,8][9,1,2,1,4][3,7,3,5,9]
ind3    [2,8][1,8][1,5]

其中每个单元格(例如 A，ind1)都有一个列表 [1,2,3]。当我将其导入数据框时:

df=pd.read_csv('filename.csv')

我得到一个与 csv 格式相同的数据框，但是，单个单元格中的列表作为字符串导入。

假设我索引了 ind1 列 A 的第一个元素df.iloc[0]['A'] 给我 '[' 而不是 1

基本上它是将 [1,2,3] 作为一个长字符串来读取，而不是作为列表来读取。

如何将所有单元格中的值转换为列表？

Answer 1

有趣的问题。如果您将使用 pandas，我会首先使用 literal_eval 对列表进行编码。

将 pandas 导入为 pd
从 ast 导入 literal_eval
df = pd.read_csv("文件名.csv")


然后您可以使用 pandas applymap 将 literal_eval 函数应用于每个单元格。

df[['A', 'B', 'C']] = df[['A', 'B', 'C']].applymap(literal_eval)

现在，您已经提取了列表，您可以像处理普通列表一样处理它们。例如这将为您提供第一个列表中的第一个元素:

df.iloc[0]['A'][0]

希望对你有帮助