有没有办法将今天的日期和'file-2‘添加到csv中的每一行？

Question

我有一个没有标题的csv文件

1,0,1,a
2,1,2,b
3,4,5,c

如何将其转换为每行包含今天日期和文件的格式？

1,0,1,a,2021-07-22,file2
2,1,2,b,2021-07-22,file2
3,4,5,c,2021-07-22,file2

这有可能做到吗？当我尝试使用csvwriter时，我只能附加到end。

with open("temp_csv.csv","a") as fout:
    writer = csv.writer(','+date+',file2\n')
    print(writer)

Answer 1

如果你需要使用CSV模块，那么这是可行的。

你的问题是你试图就地操纵csv。相反，您应该读取每一行，处理该行，然后将其写入新的输出csv文件。

import csv
from datetime import datetime

todays_date = datetime.today().strftime('%Y-%m-%d')

with open("in.csv","r") as fin, open("out.csv", 'w', newline='') as fout:
    reader = csv.reader(fin)
    writer = csv.writer(fout)
    for line in reader:
        line.append(todays_date)
        line.append("file2")
        writer.writerow(line)

Answer 2

为此，我推荐使用pandas库。

首先，使用pandas将您的文件作为dataframe df加载：

import pandas as pd

df = pd.read_csv('temp_csv.csv', header=None, index_col=0)

然后添加所需的列

df['date'] = ['2021-07-22' for i in range(len(df))]
df['file'] = ['file2' for i in range(len(df))]

最后，保存您的csv

df.to_csv('temp_csv.csv', header=False)