SounderPy

from datetime import datetime,timedelta
import sounderpy as spy

#First, decide whether you are plotting an observed sounding, or a model sounding:
observed = True
#observed = False

#Choose station:
#Note: For data query, use either use 5-digit identifier (e.g., 72518 for ALB) or station name:
station = 'KALB'

#For observed data, station name can't include the K for U.S. stations, so we handle this by removing the K:
if (observed) and len(station) == 4 and station.startswith('K'):
    station = station[1:]

#Note!  For model data, must be available from the Penn State BUFKIT site: https://www.meteo.psu.edu/bufkit/
year = 2026
month = '04' #Month, day, and hour must be "zero-padded" for model data queries, so we'll set them as strings here.
day = '01'
hour = '00'
fcsthour = '00'  #for model data

#In order to make datetime objects for naming our final sounding image, we need to convert the strings above to integers:
month_int = int(month)
day_int = int(day)
hour_int = int(hour)
fcsthour_int = int(fcsthour)
validTime = datetime(year, month_int, day_int, hour_int)

if (observed):
    timeStr = validTime.strftime("%y%m%d%H")
    print(timeStr)
    print(validTime)
    print(station)
else:
    timeStr_temp = validTime.strftime("%y%m%d%H")
    timeStr = f'{timeStr_temp}F{fcsthour}'
    print(timeStr)
    print(validTime)
    print(station)

#For observed data:
#argument list: spy.get_obs_data(station, year, month, day, hour)
if (observed):
     data = spy.get_obs_data(station,year,month,day,hour)
    
#For BUFKIT model data (can be hrrr, rap, nam, namnest, gfs, sref, hiresw)
#argument list: spy.get_bufkit_data(model, station, fcst_hour, run_year=None, run_month=None, run_day=None, run_hour=None)
else:
     data = spy.get_bufkit_data('hrrr', station, fcsthour_int, year, month, day, hour)

#Investigate data:
data

spy.build_sounding(data,save=True,filename=(f'{timeStr}_{station}.png'))