你可以按照以下方式修改代码来实现你的需求:
class InsertCsv:
def __init__(self):
self.csv_path = os.path.join(os.getcwd(), "csv")
def insert(self, df):
self.file, self.min, self.max, self.sum = df
self.year = str(self.max).split("-")[0] + ".xlsx" # 提取年份作为文件名
self.file_path = os.path.join(self.csv_path, self.year)
self.is_file_exist()
if not os.path.isfile(self.file_path):
# 文件不存在,创建新文件并写入数据
self.file.to_excel(self.file_path, sheet_name="Sheet1", index=False, encoding="utf-8")
else:
# 文件已存在,打开文件追加数据
writer = pd.ExcelWriter(self.file_path, mode='a', engine='openpyxl')
writer.book = openpyxl.load_workbook(self.file_path)
writer.sheets = dict((ws.title, ws) for ws in writer.book.worksheets)
self.file.to_excel(writer, sheet_name="Sheet1", index=False, encoding="utf-8", startrow=self.get_last_row())
writer.save()
writer.close()
def is_file_exist(self):
if not os.path.isdir(self.csv_path):
os.makedirs(self.csv_path)
def get_last_row(self):
wb = openpyxl.load_workbook(self.file_path)
sheet = wb.active # 获取活动表格
return sheet.max_row + 1
这样,将`self.max`中的日期进行拆分,提取年份作为文件名,这样同一年份的数据会被写入到对应的文件中。例如,`self.max`为"2024-01"或"2024-02"时,数据都会写入到名为"2024.xlsx"的文件中;而如果`self.max`为"2023-11",数据会写入到名为"2023.xlsx"的文件中。
以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。 |