删除一系列 csv 列

Question

我正在尝试创建一个可导入模块来删除一系列列（特别是我正在使用的文件中的第 73-177 列）。我正在尝试编辑为删除列而编写的此文件 i/o 代码基于字段名称。我想修改此代码以删除 csv 文件中的第 73-177 列。我需要做什么才能做到这一点？

def removeColumns(num1, num2, inputFILE, FileName): inPUTfile = 打开（输入文件，'r'） outPUTfile = 打开（文件名，'w'） line = inPUTfile.readline() 而线： # 删除指定的列。第一列范围编号，第二列范围编号 (+1) lineList = line.split('\t') removeCOL = "日历年" 我 = 0 而 lineList[i] != removeCOL: #(linesout?): 我 = 我 + 1 lineList.pop(i) #从list.append中删除这些字段 #写入修改的字段 删除 = "\t".join(lineList) outPUTfile.write(line) #写入新的字段名 outfile for line inPUTfile: #remove field i from each剩余行并将其写入输出文件 &修改输入行 lineList = line.split() #转换为列表 lineList.pop(i) #从列表中删除字段 line = '\t'.join(lineList) line = line + '\n' #在行尾加回车 outPUTfile.write(line)# 将修改后的行写入输出文件 inPUTfile.close() #关闭输入文件 outPUTfile.close() #关闭输出文件 返回输出PUT文件 打印出PUT文件

Answer 1

我知道您问过如何修改原始代码，但老实说，在这里我认为以不同的方式进行修改会更容易理解。 Python 有一个有用的csv 模块，它可以为您处理很多工作。比如：

import csv

remove_from = 2
remove_to = 5

with open("to_delete.csv", "rb") as fp_in, open("newfile.csv", "wb") as fp_out:
    reader = csv.reader(fp_in, delimiter="\t")
    writer = csv.writer(fp_out, delimiter="\t")
    for row in reader:
        del row[remove_from:remove_to]
        writer.writerow(row)

会转

$ cat to_delete.csv 
a   b   c   d   e   f   g
1   2   3   4   5   6   7
8   9   10  11  12  13  14
15  16  17  18  19  20  21

进入

$ cat newfile.csv 
a   b   f   g
1   2   6   7
8   9   13  14
15  16  20  21