python - 如何使用python在csv文件中重复一行n次

Question

我是 python 新手，正在尝试将多个文件合并到所需的布局中。我的一个问题涉及重复 csv 文件的每一行（不包括标题）n 次。例如

初始文件：

header1 header2 header3
abc       123    a1
def       456    b1
ghi       789    c1

修改后，文件应如下所示（例如 3 次重复）：

header1 header2 header3
abc       123    a1
abc       123    a1
abc       123    a1
def       456    b1
def       456    b1
def       456    b1
ghi       789    c1
ghi       789    c1
ghi       789    c1

使用 csv 或 pandas 在 python 中执行此操作的最佳方法是什么。如果问题太琐碎，我深表歉意，但我对使用 python 进行此类文件操作并不熟悉，并且在论坛上没有发现任何类似的问题。

谢谢，

标签： python

Answer 1

使用该csv模块，您可以执行以下操作：

import csv

with open('out.txt', 'w') as fout, open('in.txt', 'r') as fin:
        reader = csv.reader(fin)
        writer = csv.writer(fout)
        writer.writerow(next(reader))
        for l in reader:
            for i in range(3):
                writer.writerow(l)