python - 在不删除逗号的情况下将数据帧写入 csv

Question

我想将 pandas 数据框写入 csv。 df 的其中一列包含列表条目，例如[1, 2], [3, 4], ...

当我使用 df.to_csv('output.csv') 并打开输出 csv 文件时，逗号消失了。也就是说，csv 的相应列有条目 [1 2]、[3 4]、....

有没有办法在不删除这些列表中的逗号的情况下将数据框写入 csv？我也尝试过使用 csv.writer。

Answer 1

添加关键字参数quoting=csv.QUOTE_ALL:

import csv
import pandas as pd

data = {"A": [1,2,3], "B": [[11,12],[21,22],[31,32]]}
df = pd.DataFrame(data)
df.index.name='index'

# df
#        A         B
# index             
# 0      1  [11, 12]
# 1      2  [21, 22]
# 2      3  [31, 32]

df.to_csv('test.csv',quoting=csv.QUOTE_ALL)

输出:

"index","A","B"
"0","1","[11, 12]"
"1","2","[21, 22]"
"2","3","[31, 32]"