在熊猫中使用to_csv时删除一些列

Question

我有一个要写入两个文件的数据框，一个包含所有列，一个仅包含列的子集，因此对于此数据框：

 Out_data
 Out[9]: 
    A     B                 C          D       E         F
0  354   49985400          10          07   7.140899  0.212044
1  738   49985400          10          07   7.140899  0.212044
2  738   49985277          11          09   4.024423  0.098387
3  246   49985279          10          07   7.140899  0.212044

我想将其导出到两个csv文件中，一个包含所有数据，第二个仅包含来自A，B，C和D列的数据，以便csv看起来像这样：

  A,B,C,D,E,F
354,49985400,10,07,7.140899,0.212044
738,49985400,10,07,7.140899,0.212044
738,49985277,11,09,4.024423,0.098387
246,49985279,10,07,7.140899,0.212044

第二个看起来像：

  A,B,C,D
354,49985400,10,07
738,49985400,10,07
738,49985277,11,09
246,49985279,10,07

我可以使用以下方法获取第一个文件：

Out_data.to_csv(filename, mode = 'w', index=False)

我尝试使用

Out_data.to_csv(filename, mode = 'w', cols = ['A','B','C','D'] ,index=False)

但是我仍然走完全相同的输出文件吗？如何获取to_csv进行导出但删除某些列？

Answer 1

您应该将其columns用作关键字参数，而不是cols。

Out_data.to_csv(filename, mode='w', columns=['A','B','C','D'], index=False)