使用os.path.exists()方法判断文件是否存在,如果不存在则创建文件。如果存在则读取文件列表。
import os
import numpy as np
import pandas as pd
if os.path.exists('data.csv'):
with open('data.csv',mode='r',encoding='utf-8') as ff:
print("文件已经存在")
data=np.loadtxt(open("data.csv","rb"),dtype=np.str,delimiter=",").tolist()
print(data)
#print(ff.readlines())
else:
with open("data.csv", mode='w', encoding='utf-8') as ff:
data=np.loadtxt(open("data.csv","rb"),dtype=np.str,delimiter=",")
content=[]
content.append([0,0,0,0,0,0,0,0,0,0])
data=data.tolist()
data.extend(content)
data=np.array(data)
data=pd.DataFrame(data)
data.to_csv('data.csv')
print(data)
print("文件创建成功!")
不存在data.csv,初始化创建,结果如下:
存在data.csv,则打印该文件列表