python - 什么文件模式在不存在时创建新数据并在存在时追加新数据

Question

我需要执行以下操作:

1. 如果 csv 文件不存在则创建一个，如果存在则追加数据
2. 创建新的 csv 文件时，使用 dict1 的标题创建。

我的代码:

def main():
    list1 = [ 'DATE','DATASET','name1','name2','name3']
    dict1 =dict.fromkeys(list1,0)
    with open('masterResult.csv','w+b')as csvFile:
        header = next(csv.reader(csvFile))
        dict_writer = csv.DictWriter(csvFile,header,0)
        dict_writer.writerow(dict1)

if __name__ =='__main__':
    main()

Answer 1

我编写了以下示例代码，您可以引用并根据您的要求使用。首先，如果使用追加模式打开文件，则文件存在则追加，不存在则新建。现在，开始编写标题，您可以事先检查文件的大小。如果大小为零，那么显然它是一个新文件，您可以先编写 header 。如果大小不为零，则可以仅附加数据记录而不写入 header 。下面是我的示例代码。第一次运行它时，它将创建带有标题的文件。下次运行代码时，它将仅附加数据记录，而不附加标题。

 import os
 header='Name,Age'
 filename='sample.csv'
 filesize=0
 if(os.path.exists(filename) and os.path.isfile(filename)):
    filesize=os.stat(filename).st_size
 f=open(filename,'a')
 if(filesize == 0):
    f.write('%s\n' % header)
 f.write('%s\n' % 'name1,25')
 f.close()