Python 从制表符分隔的文本文件中选择多列

Question

我试图从一个文本文件中提取多列，最后得到了下面的代码，它将所有列提取17到最后。我试图找出一个正则表达式组合来告诉代码要检索哪些其他确切的列，但我有点卡住了。有人可以给我一个提示吗？f = open("datafile1.txt", "r")g = open("datafile2.txt", "w")for line in f:  if line.strip():    g.write("\t".join(line.split("\t")[17:]))f.close()g.close()

Answer 1

您可以通过简单的列表理解来实现，如下所示：

f = open("datafile1.txt", "r")

g = open("datafile2.txt", "w")

for line in f:

  if line.strip():

    g.write("\t".join([line.split("\t")[i] for i in [18, 26, 41, 45]]))

f.close()

g.close()