matplotlibストリームプロットのサブプロットカラーバーを表示

Question

私がやりたいことは簡単です：plt.subplots()を使って作成したプロットの場合、カラーバーを表示したいと思います。

だから、これは私が何をすべきかです：docsに示されているものによって

def plotVF(self, u, v): 
    m = np.sqrt(np.power(u, 2) + np.power(v, 2)) 

    xrange = np.linspace(0, u.shape[1], u.shape[1]); 
    yrange = np.linspace(0, u.shape[0], u.shape[0]); 

    x, y = np.meshgrid(xrange, yrange) 
    mag = np.hypot(u, v) 
    scale = 1 
    lw = scale * mag/mag.max() 

    f, ax = plt.subplots() 
    h = ax.streamplot(x, y, u, v, color=mag, linewidth=lw, density=3, arrowsize=1, norm=plt.Normalize(0, 70)) 
    ax.set_xlim(0, u.shape[1]) 
    ax.set_ylim(0, u.shape[0]) 
    ax.set_xticks([]) 
    ax.set_yticks([]) 
    cbar = f.colorbar(h, ax=ax) 
    cbar.ax.tick_params(labelsize=5) 

    plt.show() 

。 は、しかし、私は受信し続ける：

AttributeError: 'StreamplotSet' object has no attribute 'autoscale_None' 

をこの例では、唯一のプロットを持っていますが、私は私が直接StreamplotSetオブジェクトを返しますax.streamplotplt.colorbar().

Answer 1

を使用していない理由です、以上のものを持っています。これは、colorbarの作成に使用できるmappabpleインスタンスではありません。ただし、docs for streamplotによれば、LineCollectionと、FancyArrowPatchオブジェクトのコレクションが含まれています。 LineCollectionを使用してカラーバーを作成できます。

hからh.linesを使用してアクセスできます。このように、あなたのカラーバーを作るために、次のものが必要です。

cbar = f.colorbar(h.lines, ax=ax) 

あなたはmatplotlibギャラリー、hereにこの例を見ることができます。