给出以下熊猫DataFrame:
basket = pd.DataFrame({'id':[1,2,3], 'item':[('A', 'B',), ('C',), ('A', 'C', 'D',)]})
如何生成以下csv文件?
A,B
C
A,C,D
PS:我已经尝试过以下代码:
with open('items.csv','wb') as out:
csv_out=csv.writer(out)
for index, row in basket.iterrows():
csv_out.writerow(basket['item'])
出现此错误:
TypeError: a bytes-like object is required, not 'str'
并尝试了以下操作:
basket.to_csv("items.csv", index=False, header=False, columns=['item'])
产生的输出不是我所需要的:
"('A', 'B')"
"('C',)"
"('A', 'C', 'D')"
编辑:
目的是稍后在另一个代码中以列表列表的形式加载导出的csv。即上面给定示例中的[['A', 'B'], ['C'], ['A', 'C', 'D']]
。