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()