- 判断文件是否存在
# 如果当前股票.csv文件存在
if os.path.exists(file_root_temp):
- 去csvn中去date最后一行的数值
# 获取增量数据(code,startsdate=对应股票csv中最新日期,enddate=今天)
startdata = pd.read_csv(file_root_temp, usecols=['date'])['date'].iloc[-1]
- csv中追加数据
# 追加数据 add,请注意,如果未设置header = False,则将按原样添加标头(pandas.DataFrame的列)
data.to_csv(file_root, mode=mode, header=False)
- 删除csvn中重复的数据
# 删除重复数据,以防万一
data = pd.read_csv(file_root)
data = data.drop_duplicates(subset=['date'])
data.to_csv(file_root, index=False)
- 默认参数,调用函数的时候可以不写
def export_data(data: DataFrame, filename, type, mode=None):