首页 > 解决方案 > 在列表理解中创建/导出数据帧到 csv

问题描述

我有 4 个 DF

jeep_1 = jeep_data[jeep_data['Distance (km)'] < 5]
jeep_2 = jeep_data[(jeep_data['Distance (km)'] >= 5) & (jeep_data['Distance (km)'] <= 10)]
jeep_3 = jeep_data[(jeep_data['Distance (km)'] >= 11) & (jeep_data['Distance (km)'] <= 20)]
jeep_4 = jeep_data[jeep_data['Distance (km)'] > 21]

然后我有这个代码:

list_df = [jeep_1, jeep_2, jeep_3, jeep_4]

[item.to_csv('/Users/jacob/Desktop/Jeepney PROJECT 2/jeep_{}.csv'.format(???)) for item in list_df]

该列表是数据框的列表。

我想用不同的文件名 jeep_1.csv、jeep_2.csv 等保存它们。

如何访问数据框名称 jeep_2 的最后一个元素,或者如何实现这一点?

谢谢 !

标签: pythonpython-3.xloopslist-comprehensionexport-to-csv

解决方案


试试吧?虽然可能不是最好的解决方案......

str(my_list.index(item))

虽然这回答了你的问题,但我可能会在变量之后命名文件:

for item in my_list:
    item.to_csv('/Users/jacob/Desktop/Jeepney PROJECT 2/' + item + '.csv')

如果你想更准确,你可以导入操作系统,并使用os.path.join(your_dir, file_name, ".csv")


推荐阅读