generate_temperature_tif.py

import h5py
import numpy as np
import matplotlib.pyplot as plt
import os


variable = 'T2'
variable_lower = variable.lower()

# Directory to save PNGs
output_dir = 'data_folder/output_images_'+variable_lower
os.makedirs(output_dir, exist_ok=True)

# List of WRF output files
wrf_files = sorted([os.path.join('data_folder/wrf/', f) for f in os.listdir('data_folder/wrf/') if f.startswith('wrfout_d01')])

for i, file in enumerate(wrf_files):
    with h5py.File(file, 'r') as hdf_file:
        temperature = hdf_file[variable][0, :, :]
        temperature_flipped = np.flipud(temperature)

        plt.figure(figsize=(10, 6))
        plt.imshow(temperature_flipped, cmap='plasma')
        plt.axis('off')
        plt.savefig(os.path.join(output_dir, f'{variable_lower}_{i:03d}.png'), bbox_inches='tight', pad_inches=0)
        plt.close()