且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

如何用python删除csv文件中的某一列?代码该如何编写呢?

更新时间:2023-10-04 20:23:28

如果問題很單純,甚至連 csv 都可以不使用:

with 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 的:

import 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])

del_cvs_col 會將 fname 轉成 newfnamedelimiter 來做分割字元,且會去除掉在 idxs中指定的列。


我回答過的問題: Python-QA