csv文件中的标题每隔一行打印一次

Question

我一直在处理这个问题太久了，请向我解释为什么标题在我的输出 csv 中不断重复。

我有一个包含这些数据的输入 csv：

1. 名字，房子
2. “雅培，汉娜”，赫奇帕奇
3. “贝尔，凯蒂”，格兰芬多
4. “骨头，苏珊”，赫奇帕奇
5. “布特，特里”，拉文克劳

该问题需要颠倒姓氏和名字，将名称分成两列，并为输出 csv 创建一个包含 3 列的新标题。 这是我所拥有的：

    while True:
        try:
            # open file
            with open(sys.argv[1]) as file:
                # make reader
                reader = csv.reader(file)
                # skip first line (header row)
                next(reader)
                # for each row
                for row in reader:
                    # identify name
                    name = row[0]
                    # split at ,
                    name = name.split(", ")
                    # create var last and first, identify var house
                    last = name[0]
                    first = name[1]
                    house = row[1]


                    # writing the new csv
                    with open(sys.argv[2], "a") as after:
                        writer = csv.DictWriter(after, fieldnames=["first", "last", "house"])

                        # HEADER ONLY NEEDS TO OCCUR ONCE
                        writer.writeheader()
                        writer.writerow({"first": first, "last": last, "house": house})
                sys.exit(0)

我的输出csv：

1. 第一个，最后一个，房子
2. 汉娜、雅培、赫奇帕奇
3. 第一个，最后一个，房子
4. 凯蒂、贝尔、格兰芬多
5. 第一个，最后一个，房子
6. 苏珊，骨头，赫奇帕奇

我试过删除while循环，取消缩进和缩进，用标题名称手动写一行（这会导致错误）。 请帮忙。 谢谢！

Answer 1

您可以添加一个变量来保存是否打印了标题，例如write_header

while True:
    try:
        write_header = True
        # open file
        with open(sys.argv[1]) as file:
            # make reader
            reader = csv.reader(file)
            # skip first line (header row)
            next(reader)
            # for each row
            for row in reader:
                # identify name
                name = row[0]
                # split at ,
                name = name.split(", ")
                # create var last and first, identify var house
                last = name[0]
                first = name[1]
                house = row[1]


                # writing the new csv
                with open(sys.argv[2], "a") as after:
                    writer = csv.DictWriter(after, fieldnames=["first", "last", "house"])

                    # HEADER ONLY NEEDS TO OCCUR ONCE
                    if write_header:
                        writer.writeheader()
                        write_header = False
                    writer.writerow({"first": first, "last": last, "house": house})
            sys.exit(0)

看看我如何使用write_header

另一方面，您可以重构代码以在for循环之前打开 csv 编写器，在此处写入标头，然后像现在一样写入值，而无需每次要写入一行时都重新打开文件