Python - matplot - scatter

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

data = np.array([
    [0, 0, 1],
    [0, 1, 1],
    [0, 2, 1],
    [1, 0, 2],
    [1, 1, 3],
    [1, 2, 2],
    [2, 0, 2],
    [2, 1, 2],
    [2, 2, 2],
])

X = data[:,0]
Y = data[:,1]
Z = data[:,2]

fig = plt.figure()
ax = fig.gca(projection='3d')

ax.scatter(X, Y, Z)

plt.show()