Как мне сделать мой matplotlib 3d plot радугу и лучшую форму?

Question

Надеюсь, у тебя все хорошо. Я сделал этот простой трехмерный график, и я только начинаю с matplotlib (это мой первый трехмерный график), и я понятия не имею, как сделать его сферической формой и окрашенной в цвет радуги. Кто-нибудь может мне помочь?

Спасибо заранее. - Джо sh

        import matplotlib.pyplot as plt
        from mpl_toolkits.mplot3d import Axes3D
        import numpy as np

        fig = plt.figure()
        sub = fig.add_subplot(1, 1, 1, projection='3d')
        x = np.linspace(2,1,100)
        X, Y = np.meshgrid(x,x)
        Z = np.sin(X*X*Y*Y)
        sub.plot_surface(X, Y, Z)
        plt.show()