Python - 如何将数据写入新的 csv 文件

Question

好的，所以我对 Python 还很陌生，我有一个 .csv 文件，其中包含赛车手姓名、开始和结束时间的数据。我需要计算所花费的时间并将附加信息以及较早的数据输出到新的 .csv。

这是我目前所拥有的：

import csv
input_file = csv.DictReader(open("runners_times.csv"))
for row in input_file:
    time_taken = float(row["finish_time"])-float(row["start_time"])
    print (row)
    print (time_taken)

with open("runners_times_appended.csv", "w+") as to_file:
        writer = csv.writer(to_file, delimiter=",")
        for row in input_file:
            writer.writerows(all)

这正在创建新的 .csv，但文件为空。非常感谢您的帮助。

Answer 1

我不完全了解您如何在要保存的文件中使用 time_taken。
据我了解，您需要使用 time_taken 在 dataFrame/csvFile 中添加/更改列。你可以用pandas 包来做到这一点：

import pandas as pd
df = pd.read_csv("runners_times.csv")
df["time_taken"] = df["finish_time"] - df["start_time"]
df.to_csv("new_name.csv")

Answer 2

试试这个：

import csv
data_list = []
input_file = csv.DictReader(open("runners_times.csv"))
for row in input_file:
    time_taken = float(row["finish_time"])-float(row["start_time"])
    row.update({"time_taken":time_taken})
    data_list.append(row)
keys = data_list[0].keys()
with open("runners_times_appended.csv", "w") as fp:
    dict_writer = csv.DictWriter(fp, keys)
    dict_writer.writeheader()
    dict_writer.writerows(data_list)