#!/usr/bin/env python3 from salem import geogrid_simulator import matplotlib.gridspec as gridspec import matplotlib.pyplot as plt # plot the wrf domains over a natural earth namelist_path = 'demo_files/namelist.wps.demo' # ensure no comments in namelist g, maps = geogrid_simulator(namelist_path) fig = plt.figure(1, figsize=(5, 5)) gs = gridspec.GridSpec(1, 1) ax = fig.add_subplot(gs[0]) maps[0].set_rgb(natural_earth='lr') maps[0].visualize() gs.tight_layout(fig) plt.savefig('demo_files/wrf_domains.png', dpi=700, alpha=True, bbox_inches='tight') plt.show()
import os import salem from glob import glob import pandas as pd import numpy as np import wrf import netCDF4 as nc import matplotlib.pyplot as plt import matplotlib.dates as mdates import seaborn as sns from IQR_plot import plot_day_clm # load map settings from wps namelist fig_map, ax_map = plt.subplots(1, 1) fpath = '../wrf_input/London-3domains/namelist.wps-london-3' g, maps = salem.geogrid_simulator(fpath) maps[0].set_rgb(natural_earth='hr') maps[0].visualize(ax=ax_map, title='Domains') # export the figure fig_map.set_size_inches(6, 6) fig_map.savefig('fig/map.png') # load WRF results fl_WRF = sorted(glob('wrfout*')) ds = salem.open_mf_wrf_dataset(fl_WRF) dswrf = nc.Dataset(fl_WRF[0]) x_pos, y_pos = wrf.ll_to_xy(dswrf, latitude=[51.], longitude=[-0.1])
import xarray as xr import numpy as np import salem import matplotlib.pyplot as plt from salem.utils import get_demo_file from salem import open_wrf_dataset, get_demo_file from salem import geogrid_simulator import cartopy from salem import mercator_grid, Map, open_xr_dataset # fpath = get_demo_file('namelist_mercator.wps') fpath = 'namelist2.wps' with open(fpath, 'r') as f: # this is just to show the file print(f.read()) g, maps = geogrid_simulator(fpath) print(g) print('ss') print(maps[0]) maps[0].set_rgb(natural_earth='lr') maps[0].visualize(title='Anidamiento Qollpana Marzo 2018', addcbar=False) plt.savefig('QollpanaMarzo2018.png', dpi=300) plt.show() plt.close() #ALBBCK ds = open_xr_dataset( "/home/opti3040a/Documentos/WRF15-08-19/wrfout_d01_2018-03-05_00:00:00") grid = mercator_grid(center_ll=(-60.0, -17.9), extent=(75e5, 75e5)) smap = Map(grid, nx=500) # VAR_SSO
#%% from salem import geogrid_simulator import matplotlib.pyplot as plt if __name__ == '__main__': geo_path = '/home/zzhzhao/code/python/wrfchem_2/data2/namelist.wps' g, maps = geogrid_simulator(geo_path) maps[0].set_rgb(natural_earth='lr') # add a background image maps[0].visualize(title='Domains 1') plt.savefig('/home/zzhzhao/code/python/wrfchem_2/fig/domain.jpg', dpi=300) # %%