替换某些列的Python

Question

input.txt是制表符分隔的。

我知道一个简单的代码来替换。

import fileinput 
for line in fileinput.FileInput("input.txt",inplace=1): 
    line = line.replace("AA","0") 
    print line, 

但是，我想仅更换input.txt中（而不是整个文件input.txt中）的第3列的细胞，也想通过0替换的细胞，如果它是或AA中的任一项AAA或BB或BBB，如果不是它们中的任何一个，则将其替换为1。

在这里，我说的是“匹配整个单元格内容”

通过“匹配整个单元格内容，”我的意思是，它会被替换，只有当一个细胞（如（2,3） - 元素input.txt）完全是AA或AAA或BB或BBB。诸如“AAs”的单元不会被任何东西取代。相反，如果没有应用“匹配全部细胞内容物”，则只要细胞仅“包含”AA或AAA或BB或BBB，它就会被替换。所以一个单元格“AAhaha”将被替换为“0haha”

无论如何，重复一遍，我想替换input.txt的第三列（而不是整个文件input.txt）的单元格，我想如果它是AA或AAA或BB或BBB中的任何一个，则将单元替换为0，如果它不是其中任何一个，则以“匹配整个单元内容”方式替换单元1。

Answer 1

for line in fileinput.FileInput("input.txt",inplace=1): 
    cells = line.split('\t') 
    cells[2] = '0' if cells[2] in ('AA', 'AAA', 'BB', 'BBB') else '1' 
    print '\t'.join(cells), 

请注意，我已经对制表符分隔的数据进行了简单化处理。如果您的文件使用整个CSV/TSV格式，包含制表符和/或换行符的带引号的单元格，那么您需要csv，这是一个适当的CSV解析器。

相反，如果你想在列0细胞含有例如"a"以输出为"a"，则必须不使用csv，因为读书时，它会删除引号，而不是重新插入他们的作品。因为他们不是那个单元所需要的。

所以，首先你必须确定文件格式是如何定义的，然后你可以选择如何读写它。无论哪种方式，修改它将大致相同。

另外一个问题：我没有做任何关于linebreak的事情，所以它只会坐在最后一个单元格中。因此，如果第三个单元格是最后一个单元格，它将在单元格被替换为"0"或"1"时删除，这可能不是您想要的。虽然我们在谈论单元的数量，但如果任何一行少于3个单元，这段代码当然会抛出异常。您应该决定如何处理该问题，特别是在文本文件末尾找到空白行并不罕见。

Answer 2

你应该使用csv module此：

import csv 
with open("input.txt", "rb") as infile, open("output.txt", "wb") as outfile: 
    reader = csv.reader(infile, delimiter="\t") 
    writer = csv.writer(outfile, delimiter="\t") 
    for row in reader: 
     row[2] = "0" if row[2] in ("AAA", "AA", "BBB", "BB") else "1" 
     writer.writerow(row)