matplotlibは発散と収束の領域で合理化する

Question

私は、uとvを使って流線を描いた。発散や収束が起こっているかどうかを判定し、matplotlibでそれらの図形を同じ図形にプロットするにはどうすればよいですか？

streamline test, red is divergence and blue is convergence.

Answer 1

あなたは色をあなたが望む任意の方法で合理化し、そうしたいdivergenceのどんな形取得し、それを使用し、

import numpy as np 
import matplotlib.pyplot as plt 

Y, X = np.mgrid[-3:3:100j, -3:3:100j] 
U = -1 - X**2 + Y 
V = 1 + X - Y**2 
divUV = reduce(np.add,np.gradient(U)) + reduce(np.add,np.gradient(V)) 

fig, ax = plt.subplots() 
strm = ax.streamplot(X, Y, U, V, color=divUV, cmap=plt.cm.RdBu) 
fig.colorbar(strm.lines) 

plt.show() 

ない発散はここに見えることを確認することができますしかし、あなたはその考えを得る。また、透明度を指定してカラーメッシュをオーバーレイすることもできます。

cm = ax.pcolormesh(X, Y, divU, cmap=plt.cm.RdBu, alpha=0.4) 
fig.colorbar(cm)