from sympy import *
from spb import *
import colorcet as cc
import matplotlib.cm as cm
var("x, y")
expr = cos(x**2 + y**2)
PB.colormaps = ["solar", "aggrnyl", cm.coolwarm, cc.kbc]
plot3d(
    (expr, (x, -2, 0), (y, -2, 0)),
    (expr, (x, 0, 2), (y, -2, 0)),
    (expr, (x, -2, 0), (y, 0, 2)),
    (expr, (x, 0, 2), (y, 0, 2)),
    n = 20, backend=PB, use_cm=True
)