1import numpy as np
2import matplotlib.pyplot as plt
3import click
4import os
5import glob
6from pathlib import Path
7
8
9@click.command()
10@click.option("-i", "--input-dir", help="Input directory")
11@click.option("-o", "--output-dir", help="Output directory")
12def main(input_dir, output_dir):
13 for input_file in glob.glob(os.path.join(input_dir, "*.npz")):
14 plt.figure(figsize=(8, 8))
15 data = np.load(input_file)
16 m = data["methane"]
17 m[data["mask"]] = np.nanmedian(m)
18 plt.imshow(m)
19 plt.colorbar()
20 x = np.arange(data["methane"].shape[1])
21 y = np.arange(data["methane"].shape[0])
22 xv, yv = np.meshgrid(x, y)
23 plt.quiver(xv, yv, data["u10"], data["v10"])
24 stem = Path(input_file).stem
25 plt.savefig(os.path.join(output_dir, stem + ".png"))
26
27
28if __name__ == "__main__":
29 main()