import numpy as np
import matplotlib.pyplot as plt



def grid_scatter():
    X, Y = np.meshgrid(np.linspace(-1, 1, 10), np.linspace(-1, 1, 10))
    Z = X+2*X*Y
    plt.scatter(X, Y, c=Z)
    plt.show()

def grid_3d():
    X, Y = np.meshgrid(np.linspace(-1, 1, 100), np.linspace(-1, 1, 10))
    Z = X + 2 * X * Y
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')

    surf = ax.plot_surface(X, Y, Z, cmap='Reds')
    fig.colorbar(surf)
    plt.show()


if __name__ == "__main__":
    grid_scatter()
    grid_3d()