matplotlib（mplot3d） - 如何在3D Plot中增加轴（拉伸）的大小？

I have this so far:

到目前为止我有这个:

x,y,z = data.nonzero()    
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(x, y, z, zdir='z', c= 'red')
plt.savefig("plot.png")
x,y,z = data