Python，一起添加 CSV 文件的第 3 行和第 4 行

Question

Python 的全新用户，无论如何都不是程序员，很抱歉这个新手问题。

我有一个 CSV 文件，我需要为每一列添加第 3 行和第 4 行，最好的选择是将结果 output 到一个新文件中。

我意识到这不是一个难题，但是第一次跳入这个问题有点不知所措。

我已经搜索了一个示例程序来移植，但没有任何结果。

非常感谢您提供的任何帮助。

Answer 1

这里：

以此为例。csv：

column1,column2,column3
1,2,3
4,5,6
7,8,9
8,7,6

CSV_filename = 'example.csv'
total_columns = 3

results = []

import csv

with open(CSV_filename) as csv_file:
    temp = []
    csv_reader = csv.reader(csv_file, delimiter=',')
    line_count = 0
    for row in csv_reader:
        if line_count == 0:
            line_count += 1
        elif line_count == 3:
            for i in range(total_columns):
                temp.append(row[i])
            line_count += 1
        elif line_count == 4:
            for i in range(total_columns):
                temp_eq = temp[i]
                final_prep = int(temp_eq) + int(row[i])
                results.append(final_prep)
            line_count += 1
        else:
            line_count += 1
print(results)