readlines 从遍历数组和 writelines 到新文件

Question

bad_list 是一个从不同函数返回的数组，返回有问题需要更仔细查看的行的行号

例如array([ 1, 3, 4, 27, 50, 99], dtype=int64)

---

这个想法是读取 test.txt，并创建一个新的 test_badlines.txt，其中只包含 bad_list 中指定的有问题的行

---

这是我到目前为止所拥有的，打印行有效，但写行只在应该是 6 行时吐出 1 行

for rows in bad_list:
    filename = 'C:\\Users\\Username\\Downloads\\test.txt'
    bad_filename = str(filename)[:-4] + '_badlines.txt'
    with open(filename) as f,  open(bad_filename, 'w') as wt: 
        lines = f.readlines()
        #print lines[rows]
        wt.writelines(lines[rows])

Answer 1

lines是一个普通的list ，并且普通的list不能通过要查找的索引序列进行索引。 这里最简单的解决方案是制作一个生成器表达式来提取您关心的行，替换：

wt.writelines(lines[rows])

和：

wt.writelines(lines[row] for row in bad_list)

假设bad_list是您所描述的数组，则完全删除外部for循环； 您不想每行一遍又一遍地打开输入和输出文件，总共只打开一次。