3dbarのnp.nanの使い方がいまいちわかりません。

Question

以下のax.bar3dのzが0の値をプロットしないようにしたいのですが、以下のように、書いたら全体が真っ赤に染まってしまいました。

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import style
style.use('ggplot')

fig = plt.figure()
ax1 = fig.add_subplot(111, projection='3d')

x3 = [1,2,3,4,5,6,7,8,9,10]
y3 = [1,2,3,4,5,6,7,8,9,10]
X,Y=np.meshgrid(x3,y3)
z3 = np.zeros([10,10])
#dx = np.ones(10)
#dy = np.ones(10)
dz = np.zeros([10,10])
dz[3,3]=0
dz[5,5]=2
dz[1,7]=0
dz[5,2]=1
dz[1,0]=2
dz[dz<=0.0]=np.nan
ax1.bar3d(X.ravel(), Y.ravel(), z3.ravel(), 0.5, 0.5, dz.ravel(),color="red")

ax1.set_xlabel('x axis')
ax1.set_ylabel('y axis')
ax1.set_zlabel('z axis')

plt.show()

#エラー
 return umr_maximum(a, axis, None, out, keepdims)
 return umr_minimum(a, axis, None, out, keepdims)

どのようにmax,minを設定すれば、解決するのでしょうか？
ご指導、ご鞭撻のほど宜しくお願い致します。

Answer 1

bar3d()に nanを含んだデータの渡したときの挙動の原因はわかりませんが、単純に zの値が0より大きいときのデータのみを bar3d() の引数として渡すとよいのではないでしょうか。

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import style
style.use('ggplot')

fig = plt.figure()
ax1 = fig.add_subplot(111, projection='3d')

x3 = [1,2,3,4,5,6,7,8,9,10]
y3 = [1,2,3,4,5,6,7,8,9,10]
X,Y=np.meshgrid(x3,y3)
dz = np.zeros([10,10])
dz[3,3]=0
dz[5,5]=2
dz[1,7]=0
dz[5,2]=1
dz[1,0]=2

mask = (dz > 0.0)

ax1.bar3d(X[mask].ravel(), Y[mask].ravel(), 0.0, 0.5, 0.5, dz[mask].ravel(), color='r')

ax1.set_xlabel('x axis')
ax1.set_ylabel('y axis')
ax1.set_zlabel('z axis')

plt.show()

Comments

[Fallout_18]

解決しました！！
本当にありがとうございます。