python - Pandas 将 csv 列值读取为列表

Question

我有一个包含 6 列的大型数据框，每列都有一个列表。当我将数据帧保存到 csv 并读取 csv 时，列表将转换为字符串。

我发现一个与我接近的问题:How to read a column of csv as dtype list using pandas? 。

但问题是没有时间将每个字符串转换回列表。将所有字符串转换回列表大约需要 4-5 小时。

有一种简单的方法可以将包含列表的数据框保存为可以在 pandas 中轻松访问的格式。

Answer 1

您可以尝试使用pickle

例如:

import pandas as pd

df = pd.DataFrame({"Col": [[1,2,3], [4,5,6]]})
df.to_pickle(filename)

#Read the pickle file
df = pd.read_pickle(filename)
print(df["Col"])
print(df["Col"][0][0])

输出:

0    [1, 2, 3]
1    [4, 5, 6]
Name: Col, dtype: object
1

<强> MoreInfo