Applying a hillshade to an array using Matplotlib

matplotlib_hillshading.py

# Code via Alex Leith

from matplotlib.colors import LightSource
import matplotlib.pyplot as plt

# Sample elevation array
elevation = data_cube.elevation.isel(time=0).values

# Create hillsahde based on the array
ls = LightSource(azdeg=315, altdeg=45)
dyx = 10
shade = ls.hillshade(elevation, vert_exag=0.5, dx=dyx, dy=dyx, fraction=1.0)

# Modify the original elevation data to include shade
data_cube['shaded'] = (['latitude', 'longitude'], elevation * shade)

# Plot using bilinear interpolation to remove artefacts
data_cube.shaded.plot(size=10, interpolation='bilinear')

Thanks!