从文件中提取特定列以在python中创建新文件

Question

我有一个看起来像这样的文件：

o 345644 0 0 0 1
0 454545 0 0 2 2
0 423233 0 0 1 1
. .      . . . .
. .      . . . .
. .      . . . .

我想做的是提取第二列和第六列以创建一个文本文件。 我编写了以下代码，但是我得到的输出文件仅包含第二列。 我想要的输出是一个看起来像这样的新文件：

newfile.txt

 345644 1
 454545 2
 423233 1
 .      .
 .      .
 .      .

这是我的代码：

dta_2path = open("file.txt","r")
ws_2= open("newfile.txt", "w")

for line in dta_2path:
    if line.strip():
       ws_2.write("\t".join(line.split()[1:2]+ line.split()[9:10])+"\n") 

dta_2path.close()
ws_2.close()

任何有关如何解决此问题的帮助将不胜感激。 提前致谢！

Answer 1

尝试这个：

for line in dta_2path:
    if line.strip():
        cols = line.split()
        ws_2.write(cols[1] + "\t" + cols[5] + "\n")

Answer 2

您选择了错误的索引。 如果您使用line.split(" ")[1]和line.split(" ")[5]它应该可以工作。