瀑布圖python？

Question

是否有一個python模塊可以像MATLAB那樣做瀑布圖？我搜索了'numpy瀑布'，'scipy瀑布'和'matplotlib瀑布'，但沒有找到任何東西。

Answer 1

在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() 

我不知道如何得到的結果作爲Matlab的確實不錯。

---

如果您想了解更多，您也可以看看MayaVi：http://mayavi.sourceforge.net/

Answer 2

您可以在matplotlib使用PolyCollection類做的瀑布。請參閱此特定example以瞭解如何使用此課程進行瀑布的更多詳細信息。

此外，您可能會發現這個blog post很有用，因爲作者表示在某些特定情況下（取決於所選視角），您可能會獲得某些「視覺錯誤」。

下面是與matplotlib（從博客文章圖像）發了瀑布的示例： image http://austringer.net/images/biosonar/wfall_demo.png

Answer 3

維基百科類型的Waterfall chart一個也可以得到這樣的：

import numpy as np 
import pandas as pd 

def waterfall(series): 
    df = pd.DataFrame({'pos':np.maximum(series,0),'neg':np.minimum(series,0)}) 
    blank = series.cumsum().shift(1).fillna(0) 
    df.plot(kind='bar', stacked=True, bottom=blank, color=['r','b']) 
    step = blank.reset_index(drop=True).repeat(3).shift(-1) 
    step[1::3] = np.nan 
    plt.plot(step.index, step.values,'k') 

test = pd.Series(-1 + 2 * np.random.rand(10), index=list('abcdefghij')) 
waterfall(test)