1 import numpy as np
2 import matplotlib.pyplot as plt
3 from mpl_toolkits.mplot3d import Axes3D # used for 3d plot
4
5 x = np.linspace(0,100,50)
6 y = np.linspace(0,100,50)
7 x,y = np.meshgrid(x,y)
8 z = x**2 + y**2
9
10 figure1 = plt.figure() # create a figure
11 ax = figure1.gca(projection='3d') # define figure as 3d
12 ax.scatter(x,y,z) # define the three axes
13 ax.set_xlabel('X index')
14 ax.set_ylabel('Y index')
15 ax.set_zlabel('Z index')
16
17 plt.savefig('fig1.png',format='png',dpi=100)
