値が０のところを取り除きたい。

Question

以下のような2次元配列の値をマッピングしたときに、

p_map=[[0.         0.         0.         0.         0.00390625 0.           0.         0.         0.        ]
 [0.         0.         0.         0.0390625  0.         0.0390625    0.         0.         0.        ]
 [0.         0.         0.0703125  0.         0.015625   0.           0.0703125  0.         0.        ]
 [0.         0.0390625  0.         0.078125   0.         0.078125     0.         0.0390625  0.        ]
 [0.00390625 0.         0.015625   0.         0.015625   0.           0.015625   0.         0.00390625]
 [0.         0.0390625  0.         0.078125   0.         0.078125     0.         0.0390625  0.        ]
 [0.         0.         0.0703125  0.         0.015625   0.           0.0703125  0.         0.        ]
 [0.         0.         0.         0.0390625  0.         0.0390625    0.         0.         0.        ]
 [0.         0.         0.         0.         0.00390625 0.           0.         0.         0.        ]]

import matplotlib.pyplot as plt

plt.imshow(p_map,cmap="cool")
plt.colorbar()
plt.show()

としてやれば、

となりますが、このときに0の値を除いて、０以上の値だけをプロットしたいのですが（背景色を例えば黒に指定してプロットしたい）0だけを除くことはこの平面プロットだと不可能なのでしょうか？

3次元プロットの時は、

fig = plt.figure()
ax = Axes3D(fig, rect=(0.1,0.1,0.8,0.8))
.....
ax.w_xaxis.set_pane_color((0, 0, 0, 0))
ax.w_yaxis.set_pane_color((0, 0, 0, 0))
ax.w_zaxis.set_pane_color((0, 0, 0, 1))
###
mask = p_map>0.0
ax.bar3d(X[mask].ravel(),Y[mask].ravel(), 0, 0.2,0.2,  p_map[mask].ravel() ,color=clrs)

などのようにしてやればよかったのですが、、、
今回はデータそのものが平面として表れるので、０を消すのは厳しい（？）。
ご教授いただければ幸いです。

Comments

[hayataka2049]

それっぽい表示になるカラーマップを使えば良いのではないでしょうか

[Fallout_18]

？

[hayataka2049]

この中で左側が背景色にしたい色に近くて、全体的な色調が好みのものを　https://matplotlib.org/tutorials/colors/colormaps.html

[Fallout_18]

私の質問はそうではないと思うのですが。。

Answer 1

cmap を弄って最小値の時のα値を0(透明)にするってのはどうでしょう

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

cmap = cm.cool
cmap_data = cmap(np.arange(cmap.N))
cmap_data[0, 3] = 0 # 0 のときのα値を0(透明)にする
customized_cool = colors.ListedColormap(cmap_data)

plt.imshow(p_map,cmap=customized_cool)
plt.colorbar()
plt.show()

Comments

[Fallout_18]

ありがとうございます、参考にさせて頂きます！