Waterfall plot python?

Question

Is there a python module that will do a waterfall plot like MATLAB does? I googled \'numpy waterfall\', \'scipy waterfall\', and \'matplotlib waterfall\', but did not find a

Answer 1

Have a look at mplot3d:

# copied from 
# http://matplotlib.sourceforge.net/mpl_examples/mplot3d/wire3d_demo.py

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X, Y, Z = axes3d.get_test_data(0.05)
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)

plt.show()

I don't know how to get results as nice as Matlab does.

---

If you want more, you may also have a look at MayaVi: http://mayavi.sourceforge.net/