python - 避免将列表存储为 pandas 中的字符串

Question

我有一个 pandas 数据框，其中包含一个包含列表的列

df = pd.DataFrame({'ID': ['ID1', 'ID2'], 
                    'colA': [['AB', 'CD'], ['AB']]})
df

    ID  colA
0   ID1 [AB, CD]
1   ID2 [AB]

当我保存在数据框中时，它将具有多个值的列表转换为字符串，并用引号保留单个值

ID, colA
ID1, "['AB', 'CD']"
ID2,['AB']

第二行未存储为字符串，因为它仅包含列表中的单个值。

当我再次读取此 csv 时，我遇到问题，因为它将数据转换为:

    ID  colA
0   ID1 ['AB', 'CD']
1   ID2 ['AB']

如何避免这种情况？我想这样读取我的数据

    ID  colA
0   ID1 [AB, CD]
1   ID2 [AB]

Answer 1

to_json

df.to_json('my.json')
pd.read_json('my.json')

    ID      colA
0  ID1  [AB, CD]
1  ID2      [AB]