从python3中的csv文件中删除特定列

Question

需要删除多个csv文件的某些特定列和行（按索引），而不创建新文件。

对于下面的代码，它在每行之后给出带有新空白行的输出。

import csv

with open('file.csv') as fd:
   reader = csv.reader(fd)
   valid_rows = [row for idx, row in enumerate(reader) if idx != 0]

with open('file.csv', 'w') as out:
   csv.writer(out).writerows(valid_rows)

什么是更简单的方法（可能是其他 python 库）？

Answer 1

由于您不想生成任何新的 csv 文件并希望数据执行操作，我建议您使用Pandas Framework。在这个框架中使用drop函数。

考虑以下示例：

Sample.csv：

col1,col2,col3,col4
1,2,3,4
5,6,7,8
9,10,11,12
13,14,15,16
17,18,19,20

代码：

import pandas as pd
df = pd.read_csv('./Sample.csv')

删除列：

df.drop('col3', axis = 1, inplace = True)

df 内容：

   col1     col2    col4
0   1   2   4
1   5   6   8
2   9   10  12
3   13  14  16
4   17  18  20

删除行：

df.drop(df.index[[1,4]], inplace = True)

df 内容：

   col1     col2    col4
0   1   2   4
2   9   10  12
3   13  14  16

最后保存编辑后的csv文件：

df.to_csv('new_sample.csv', index = False)