前言
基于一个需求，希望根据用户输入一个起报时间、一个预报时间、一个时间间隔，读取对应的wrf模式数据进行绘图。举个例子如下：
起报时间为：2022071000 预报时常为：24h 时间间隔为：3h一次
实现函数：datetime.datetime
主要实现代码如下所示：
import sys import datetime date = sys.argv[1] date = str(date) frst = sys.argv[2] step = sys.argv[3] path = r'/Users/WRF/outdata/'+ date #只能是已经存在的文件目录且有数据才可以进行读取 start = datetime.datetime.strptime(date,'%Y%m%d%H').strftime("%Y-%m-%d_%H:%M:%S") end = (datetime.datetime.strptime(date,'%Y%m%d%H')+datetime.timedelta(hours=int(frst))).strftime("%Y-%m-%d_%H:%M:%S") intp = (datetime.datetime.strptime(date,'%Y%m%d%H')+datetime.timedelta(hours=int(step))).strftime("%Y-%m-%d_%H:%M:%S") fstart = path+'/wrfout_d01_'+start+'*' fintp = path+'/wrfout_d01_'+intp+'*' fend = path+'/wrfout_d01_'+end+'*' file = path+'/*' filestart = glob.glob(fstart) fileintp = glob.glob(fintp) fileend = glob.glob(fend) filelist = glob.glob(file) filelist.sort() rstart = np.array(np.where(np.array(filelist)==filestart))[0][0] rintp = np.array(np.where(np.array(filelist)==fileintp))[0][0] rend = np.array(np.where(np.array(filelist)==fileend))[0][0] fn = filelist[rstart:rend:rintp]
最终，这个fn中就是我们所需要的数据文件名，下面进行循环读取即可。
