修改DataFrame导出CSV代码,实现周期分层式可变文件名
调整CSV导出文件名实现周期+分层命名格式
你只需要基于循环变量i,通过整数除法和取余运算算出对应的周期数和分层数,再替换to_csv里的文件名即可,原循环框架完全保留。
修改后的代码如下:
for i in range(round(len(usersDf)/577)): # 计算周期数:每5个分层为一个周期,从1开始计数 period_num = (i // 5) + 1 # 计算当前周期内的分层数,从1开始计数 layer_num = (i % 5) + 1 # 生成目标文件名并导出 usersDf.loc[i*577:(i+1)*577,:].to_csv(f'Period {period_num} layer {layer_num}.csv')
逻辑说明
- 周期计算:
i // 5会将每5个连续的i值归为一组(比如i=0-4时结果为0,i=5-9时结果为1),加1后得到从1开始的周期序号 - 分层计算:
i % 5会得到当前组内的偏移量(0-4),加1后得到当前周期内从1开始的分层序号 - 文件名用f-string格式化,语法简洁直观;如果习惯字符串拼接,也可以写成
'Period ' + str(period_num) + ' layer ' + str(layer_num) + '.csv',效果完全一致
内容的提问来源于stack exchange,提问作者Matthew Ford
相关产品推荐
相关产品推荐

