Python读取两个文件，每个轮流读一行写入一个新文件

一、需求

现有两个文件：文件 A 和文件 B，要读一行文件 A ，写入新文件 C，再读一行文件 B，写入文件 C；之后再重新读一行文件 A 写入 C，再读一行文件 B 写入 C，直到读完文件 A 和 B。

二、代码

def write_file():
    with open('a.txt','r') as f1, open('b.txt','r') as f2, open('c.txt','w') as wp:
        while f1 or f2:
            if f1:
                wp.write(f1.readline())
            if f2:
                wp.write(f2.readline())

以上，问题解决~