一、问题描述
我有47个csv气象数据文件,格式相同,我需要提取每个表的第5列,即如图“combination”列,利用for循环批量如何进行呢?
二、思路&代码
#导包
import pandas as pd
#a列表将集合了47个csv文件的dataframe,可以用a[i]的思路来提取任意dataframe
a=[]
for i in range(0,47):
path = "填入你的路径"+ str(file_list[i])+"_TEM.csv" #file_list是站点序号集合的list
locals()["dataframe_"+ str(file_list[i])]=pd.read_csv(path,encoding='gbk')#locals()循环给每个dataframe创建动态变量名称,格式即"dataframe_对应的站点序号"
a.append(locals()["dataframe_"+ str(file_list[i])])#逐个添加每个dataframe
locals()["datelist_5_"+ str(file_list[i])] = a[i].iloc[:, 4]#等号左侧同样是动态命名变量,等号右侧获取每个dataframe的第5列,并命名为"datalist_5_对应的站点序号"的格式
#print(dataframe_56376)#测试dataframe数据动态变量是否命名成功,并可以成功提取
print(a[46])
#提取第五列
print(datelist_5_56376)#测试第5列数据动态变量是否命名成功,并可以成功提取
结果:
成功!
三、收获
①用local()函数动态命名
原因是等式左侧的变量命名时不能添加运算符,否则会报错为SyntaxError: cannot assign to operator。
②使用创建list的思想
从前都是append简单的对象,比如一个变量、一个数字等;本次操作将dataframe作为list的组成部分,按序号可以轻松提前对应元素。
谢谢阅读~