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

问题描述

问 题

如何用python删除csv文件中的某一列？比如名为a.csv中的数据：

index A B C
0 1 3 5
1 2 4 6
2 7 8 9

想要删除第二列（B列）该如何删除呢？代码该如何编写呢？
求大神！万分感谢！！！

解决方案

如果問題很單純，甚至連 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 轉成 newfname 用 delimiter 來做分割字元，且會去除掉在 idxs中指定的列。

---

我回答過的問題: Python-QA

这篇关于如何用python删除csv文件中的某一列？代码该如何编写呢？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！