python之在 Pandas 中使用 to_csv 时删除一些列_编程设计

2025年05月04日13Leo_wl

我有一个数据框，我想将其写入两个文件，一个包含所有列，一个只包含列的子集所以对于这个数据框:

 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 导出但删除一些列？

请您参考如下方法:

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

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