import spharpy
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec
coords = spharpy.samplings.equal_area(n_max=0, n_points=500)
data = np.sin(coords.colatitude) * np.cos(coords.azimuth)
fig = plt.figure(figsize=(9, 7))
gs = GridSpec(nrows=1, ncols=2, width_ratios=[20, 1], wspace=0.3)
ax = fig.add_subplot(gs[0], projection='3d')
cax = fig.add_subplot(gs[1])
spharpy.plot.balloon(coords, data, ax=[ax, cax])
plt.show()
