之前的博客中推荐了几个在线计算TEC等参数的网站，很方便，但是每次计算还需要手动一个一个在网页上输入信息，稍有一些麻烦，因此博主利用Python实现了一个在本地输入信息，Python自动填充到网页并打印出结果的程序
计算源码
# -*- coding: utf-8 -*-# @Time : /1/5 14:21# @Author : xymeng# @FileName: 002.py 利用IRI模型在网页上计算TEC等参数专用程序# @Software: PyCharmfrom selenium import webdriverfrom selenium.webdriver.chrome.service import Servicefrom mon.keys import Keysimport timeimport jsonfrom mon.by import Byfrom selenium.webdriver.support.ui import Selectimport pandas as pdimport io'''定义数据容纳列表'''TEC = []IV = [] #Independent Variable'''定义基本信息'''year = input('请输入计算年份：'.format())month = input('请输入计算月份：'.format())month = month.zfill(2)Lat = input('请输入要计算的纬度(-90. - 90. deg.)：'.format())Lon = input('请输入要计算的经度(0. - 360. deg.)：'.format())Lat = Lat + '.'Lon = Lon + '.'pf = input('请输入配置文件类型：\n1:Height [60. - 2000. km] 2:Latitude [-90. - 90. deg.] 3:Longitude [0. - 360. deg.] 4:Year [1958-] 5:Month [1-12] 6:Day of month[1-31] 7:Day of Year[1-365] 8:Hour profile[0.-24.]：\n'.format())st = input('请输入计算起始度数：'.format())sp = input('请输入计算截至度数：'.format())interval = input('请输入度数间隔：'.format())Upperboundary = input('请输入TEC最上层界限 (110. - 10000. km)：'.format())Upperboundary = Upperboundary + '.'s = Service("E:\chromedriver_win32\chromedriver.exe") #这里写本地的msedge的所在路径driver = webdriver.Chrome(service=s)driver.get("https://ccmc.gsfc.nasa.gov/modelweb/models/iri_vitmo.php") #该处为具体网址driver.refresh() #刷新页面driver.maximize_window() #浏览器最大化'''以下代码是输入计算的基本信息'''driver.find_element(By.NAME,'year').clear()driver.find_element(By.NAME,'year').send_keys(year)Select(driver.find_element(By.NAME,'month')).select_by_value(month)driver.find_element(By.NAME,'latitude').clear()driver.find_element(By.NAME,'latitude').send_keys(Lat)driver.find_element(By.NAME,'longitude').clear()driver.find_element(By.NAME,'longitude').send_keys(Lon)Select(driver.find_element(By.NAME,'profile')).select_by_value(pf)driver.find_element(By.NAME,'start').clear()driver.find_element(By.NAME,'start').send_keys(st)driver.find_element(By.NAME,'stop').clear()driver.find_element(By.NAME,'stop').send_keys(sp)driver.find_element(By.NAME,'step').clear()driver.find_element(By.NAME,'step').send_keys(interval)driver.find_element(By.NAME,'htec_max').clear()driver.find_element(By.NAME,'htec_max').send_keys(Upperboundary)driver.find_element(By.NAME,'height').clear()driver.find_element(By.NAME,'height').send_keys(450.)'''以下是选中和取消独立变量'''driver.find_element(By.XPATH,'//input[@value="06"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="07"]').send_keys(Keys.SPACE) # send space'''以下代码分别取消选中输出值复选框，Electron_density (Ne), m-3 ， Ratio of Ne and F2 peak density(Ne/NmF2)>'''driver.find_element(By.XPATH,'//input[@value="17"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="18"]').send_keys(Keys.SPACE) # send space'''以下代码分别选中输出值复选框，ION Temperature , TEC , NmF2 , foF2'''driver.find_element(By.XPATH,'//input[@value="20"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="29"]').send_keys(Keys.SPACE) # send spacedriver.find_element(By.XPATH,'//input[@value="35"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="44"]').send_keys(Keys.SPACE) # send spacecount = 4 #所选输出参数值'''点击计算'''driver.find_element(By.XPATH,'//input[@value="Submit"]').click() # send space'''以下为提取计算结果'''result = driver.find_element(By.TAG_NAME,'pre')data = result.text'''以下为将计算结果先保存到txt文本中'''path = r'F:\IRI计算包\result\002.txt'with open(path,'w') as xym:xym.write(data)'''以下为将计算的结果从保存到的txt文本中提取出来'''with open(path,'r') as yxc:findata = yxc.readlines()tecnum = 0 #TEC输出值在第几个tecvalnum = 0 #TEC输出值从第几行开始for a in range(len(findata)):if findata[a][2:5] == 'TEC':tecnum = aif findata[a] == '\n':tecvalnum = a + 2for b in range(tecvalnum,len(findata)):tecdata = findata[b].split(' ')for c in range(tecdata.count('')):tecdata.remove('')TEC.append(float(tecdata[tecnum-1]))IV.append(float(tecdata[0]))print(TEC)print(IV)
绘图源码
# -*- coding: utf-8 -*-# @Time : /1/21 15:13# @Author : xymeng# @FileName: 002-plot.py 用于绘制IRI模型计算出的TEC曲线# @Software: PyCharmimport matplotlib.pyplot as pltfrom selenium import webdriverfrom selenium.webdriver.chrome.service import Servicefrom mon.keys import Keysimport timeimport jsonfrom mon.by import Byfrom selenium.webdriver.support.ui import Selectimport pandas as pdimport ioimport matplotlib.pyplot as pltimport numpy as np'''定义数据容纳列表'''TEC = []IV = [] #Independent Variable'''定义基本信息'''year = input('请输入计算年份：'.format())month = input('请输入计算月份：'.format())month = month.zfill(2)Lat = float(input('请输入要计算的纬度(-90. - 90. deg.)：'.format()))Lon = float(input('请输入要计算的经度(0. - 360. deg.)：'.format()))# Lat = Lat + '.'# Lon = Lon + '.'pf = input('请输入配置文件类型：\n1:Height [60. - 2000. km] 2:Latitude [-90. - 90. deg.] 3:Longitude [0. - 360. deg.] 4:Year [1958-] 5:Month [1-12] 6:Day of month[1-31] 7:Day of Year[1-365] 8:Hour profile[0.-24.]：\n'.format())st = input('请输入计算起始度数：'.format())sp = input('请输入计算截至度数：'.format())interval = input('请输入度数间隔：'.format())Upperboundary = input('请输入TEC最上层界限 (110. - 10000. km)：'.format())Upperboundary = Upperboundary + '.'s = Service("E:\chromedriver_win32\chromedriver.exe") #这里写本地的msedge的所在路径driver = webdriver.Chrome(service=s)driver.get("https://ccmc.gsfc.nasa.gov/modelweb/models/iri_vitmo.php") #该处为具体网址driver.refresh() #刷新页面driver.maximize_window() #浏览器最大化'''以下代码是输入计算的基本信息'''driver.find_element(By.NAME,'year').clear()driver.find_element(By.NAME,'year').send_keys(year)Select(driver.find_element(By.NAME,'month')).select_by_value(month)driver.find_element(By.NAME,'latitude').clear()driver.find_element(By.NAME,'latitude').send_keys(Lat)driver.find_element(By.NAME,'longitude').clear()driver.find_element(By.NAME,'longitude').send_keys(Lon)Select(driver.find_element(By.NAME,'profile')).select_by_value(pf)driver.find_element(By.NAME,'start').clear()driver.find_element(By.NAME,'start').send_keys(st)driver.find_element(By.NAME,'stop').clear()driver.find_element(By.NAME,'stop').send_keys(sp)driver.find_element(By.NAME,'step').clear()driver.find_element(By.NAME,'step').send_keys(interval)driver.find_element(By.NAME,'htec_max').clear()driver.find_element(By.NAME,'htec_max').send_keys(Upperboundary)driver.find_element(By.NAME,'height').clear()driver.find_element(By.NAME,'height').send_keys(450.)'''以下是选中和取消独立变量'''driver.find_element(By.XPATH,'//input[@value="04"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="07"]').send_keys(Keys.SPACE) # send space'''以下代码分别取消选中输出值复选框，Electron_density (Ne), m-3 ， Ratio of Ne and F2 peak density(Ne/NmF2)>'''driver.find_element(By.XPATH,'//input[@value="17"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="18"]').send_keys(Keys.SPACE) # send space'''以下代码分别选中输出值复选框，ION Temperature , TEC , NmF2 , foF2'''driver.find_element(By.XPATH,'//input[@value="20"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="29"]').send_keys(Keys.SPACE) # send spacedriver.find_element(By.XPATH,'//input[@value="35"]').click() # clickdriver.find_element(By.XPATH,'//input[@value="44"]').send_keys(Keys.SPACE) # send spacecount = 4 #所选输出参数值'''点击计算'''driver.find_element(By.XPATH,'//input[@value="Submit"]').click() # send space'''以下为提取计算结果'''result = driver.find_element(By.TAG_NAME,'pre')data = result.text'''以下为将计算结果先保存到txt文本中'''path = r'F:\IRI计算包\result\002.txt'with open(path,'w') as xym:xym.write(data)'''以下为将计算的结果从保存到的txt文本中提取出来'''with open(path,'r') as yxc:findata = yxc.readlines()tecnum = 0 #TEC输出值在第几个tecvalnum = 0 #TEC输出值从第几行开始for a in range(len(findata)):if findata[a][2:5] == 'TEC':tecnum = aif findata[a] == '\n':tecvalnum = a + 2for b in range(tecvalnum,len(findata)):tecdata = findata[b].split(' ')for c in range(tecdata.count('')):tecdata.remove('')TEC.append((float(tecdata[tecnum-1]))*10)IV.append(float(tecdata[0]))time = [0,2,4,6,8,10,12,14,16,18,20,22,24]plt.plot(time,TEC,'k*-',markersize=10)plt.xlabel('Hour')plt.ylabel('TEC, 10^16 m-2')plt.title('IRI')plt.show()
