web-dev-qa-db-ja.com

Matplotlib:等高線図の上にフォアグラウンドに散布図

散布図をmatplotlibの前景に持ってくる方法を知っている人はいますか?等高線の上に散布図を表示する必要がありますが、デフォルトでは下にプロットされます...

前もって感謝します!

30
Mike

zorder パラメータを使用して、さまざまなプロットを表示する順序を手動で選択できます。 scatterメソッド。

例として、左側のサブプロットの散布図がzorder=1と右側のサブプロットにはzorder=-1。最も高いzorderのオブジェクトが上に配置されます。つまり、最初のサブプロットでは等高線の上に散布図が配置され、2番目のサブプロットの下では散布図が配置されます。

import numpy as np
import matplotlib.cm as cm
import matplotlib.mlab as mlab
import matplotlib.pyplot as plt

delta = 0.025
x = np.arange(-3.0, 3.0, delta)
y = np.arange(-2.0, 2.0, delta)
X, Y = np.meshgrid(x, y)
Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
Z = 10.0 * (Z2 - Z1)

norm = cm.colors.Normalize(vmax=abs(Z).max(), vmin=-abs(Z).max())
cmap = cm.PRGn

levels = np.arange(-2.0, 1.601, 0.4)

fig, axes = plt.subplots(1,2, sharey=True)

for ax, zord in Zip(axes, [1, -1]):
    ax.contourf(X, Y, Z, levels,
                cmap=cm.get_cmap(cmap, len(levels)-1),
                norm=norm)
    ax.autoscale(False) # To avoid that the scatter changes limits
    ax.scatter(np.random.uniform(-3,3,10),
               np.random.uniform(-2,2,10),
               zorder=zord)
    ax.set_title('Scatter with zorder={0}'.format(zord))

enter image description here

36
sodd