Note
Go to the end to download the full example code.
HealPIX Image visualization#
HealPIX maps can be viewed as a 2D image rotated by 45 deg. This is useful for quick visualization with image viewers without distorting the native pixels of the image.
import matplotlib.pyplot as plt
import numpy as np
import torch
from matplotlib.colors import Normalize
from PIL import Image
from earth2grid.healpix import Grid
grid = Grid(level=8)
lat = torch.tensor(grid.lat)
lat_img = grid.to_image(lat)
# Use Image to save at full resolution
normalizer = Normalize(vmin=np.nanmin(lat_img), vmax=np.nanmax(lat_img))
array = normalizer(lat_img)
array = plt.cm.viridis(array)
array = (256 * array).astype("uint8")
# set transparency for nans
array[..., -1] = np.where(np.isnan(lat_img), 0, 255)
image = Image.fromarray(array)
image.save("hpx_grid.png")
/Users/nbrenowitz/workspace/nvidia/earth2grid/examples/sphinx_gallery/hpx2grid.py:44: DeprecationWarning: __array_wrap__ must accept context and return_scalar arguments (positionally) in the future. (Deprecated NumPy 2.0)
array[..., -1] = np.where(np.isnan(lat_img), 0, 255)
Total running time of the script: (0 minutes 0.165 seconds)