python将dataframe写入csv_从DataFrame python pandas写入csv

我写了一个程序，在其中我添加了两列并将答案写入CSV文件，但是当我只想编写列的选择时，我得到了错误。

我的逻辑是：import pandas as pd

df = pd.DataFrame({'A' : ['foo', 'bar', 'foo', 'bar',

'foo', 'bar', 'foo', 'bar'],

'B' : ['one', 'one', 'two', 'two',

'two', 'two', 'one', 'two'],

'C' : [56, 2, 3, 4, 5, 6, 0, 2],

'D' : [51, 2, 3, 4, 5, 6, 0, 2]})

grouped = df.groupby(['A', 'B']).sum()

grouped['sum'] = (grouped['C'] / grouped['D'])

# print (grouped[['sum']])

a = pd.DataFrame(grouped)

a.to_csv("C:\\Users\\test\\Desktop\\test.csv", index=False, cols=('A','B','sum'))

我怎么能只写A列，B列和Sum列的数据呢。

我得到以下错误Traceback (most recent call last):

File "C:\Users\test\Desktop\eclipse\yuy\group.py", line 19, in

a.to_csv("C:\\Users\\test\\Desktop\\test.csv", index=False, cols=('A','B','sum'))

File "C:\Python27\lib\site-packages\pandas\core\frame.py", line 1126, in to_csv

date_format=date_format)

File "C:\Python27\lib\site-packages\pandas\core\format.py", line 992, in __init__

self.obj = self.obj.loc[:, cols]

File "C:\Python27\lib\site-packages\pandas\core\indexing.py", line 1018, in __getitem__

return self._getitem_tuple(key)

File "C:\Python27\lib\site-packages\pandas\core\indexing.py", line 595, in _getitem_tuple

self._has_valid_tuple(tup)

File "C:\Python27\lib\site-packages\pandas\core\indexing.py", line 106, in _has_valid_tuple

if not self._has_valid_type(k, i):

File "C:\Python27\lib\site-packages\pandas\core\indexing.py", line 1100, in _has_valid_type

(key, self.obj._get_axis_name(axis)))

KeyError: "[['A', 'B', 'sum']] are not in ALL in the [columns]"