如何用python删除csv文件中的某一列?比如名为a.csv中的数据:
index A B C
0 1 3 5
1 2 4 6
2 7 8 9
想要删除第二列(B列)该如何删除呢?代码该如何编写呢?
求大神!万分感谢!!!
如果問題很單純,甚至連
1csv
都可以不使用:
1
2
3
4with open('old.csv') as reader, open('new.csv', 'w') as writer:
for line in reader:
items = line.strip().split()
print(' '.join(items[:2]+items[3:]), file=writer)
寫個 general 的:
1
2
3
4
5
6
7
8
9
10import csv
def del_cvs_col(fname, newfname, idxs, delimiter=' '):
with open(fname) as csvin, open(newfname, 'w') as csvout:
reader = csv.reader(csvin, delimiter=delimiter)
writer = csv.writer(csvout, delimiter=delimiter)
rows = (tuple(item for idx, item in enumerate(row) if idx not in idxs) for row in reader)
writer.writerows(rows)
del_cvs_col('a.csv', 'b.csv', [2])
1del_cvs_col
會將
1fname
轉成
1newfname
用
1delimiter
來做分割字元,且會去除掉在
1idxs
中指定的列。
我回答過的問題: Python-QA
仅供参考,感觉好像有点问题
1
2
3
4
5
6
7
8
9
10
11
12# coding=utf-8
import csv
origin_f = open('csv.csv', 'rb')
new_f = open('new_csv.csv', 'wb+')
reader = csv.reader(origin_f)
writer = csv.writer(new_f)
for row in reader:
del row[-1]
print row
writer.writerow(row)
origin_f.close()
new_f.close()