python分割数据库并保存为新的文件

img

这需要分析原数据结构，才好针对性地提出解决方案，思路是：用pandas读取成dataframe,用行列索引提取相应数据写入不同文件。

这个问题应该不复杂，提供一个思路：
1、读取文件Grid_solar_EV_data.txt

import pandas as pd

file_dir='E:/data/' ##换成你自己的位置
data=pd.read_csv(file_dir+'Grid_solar_EV_data.txt',sep=',') ##假设分隔符为逗号

2、按照城市名称筛选数据

city_data=data[data['city']=='NY'] ##假设城市字段名为city，以NY为例

3、将筛选出来的数据保存为txt文件

city_data.to_csv('1-NY.txt')

由于三个城市的数据只是城市名不同，所以可以写成一个函数。

def save_txt(data,city_name):
    city_data=data[data['city']==city_name]
    city_data.to_csv('1-'+city_name+'.txt')

##调用函数
save_txt(data,'NY')

如果原文件的DateTime列不在第一列，还需要处理一下列的位置。

参考：