特定のピクセル値の部分を指定した色で塗る方法

グレースケール画像があります。ピクセルが「0, 10, 20, 30, 40, 50」のカラーマップのように特定のピクセルのみで構成されています。

python
1array([[ 0,  0,  0, ...,  0,  0,  0],
2       [ 0,  0,  0, ...,  0,  0,  0],
3       [ 0,  0,  0, ...,  0,  0,  0],
4       ...,
5       [20, 20, 20, ...,  0,  0,  0],
6       [10, 10, 10, ...,  0,  0,  0],
7       [30, 30, 30, ...,  0,  0,  0]], dtype=uint8)
8
9gray_img = cv2.imread("test.jpg", 0)
10print(gray_img.shape)
11>>>(9702, 9702)
12print(np.unique(gray_img)) # pixcel数の種類
13>>>[0, 10, 20, 30, 40, 50]
14
15X = np.zeros(gray_img.shape, np.float32)
16#　10,50を白、　それ以外を黒にしたい

この画像の10と50の部分を他と区別するため、
・10と50を白
・それ以外を黒
にする方法はないでしょうか？

np.zeros()で行列を作ってから代入するのはたまに聞くのですが、具体的なやり方がわからないためご教授お願いします

行動規範の内容に同意します

回答1件

ベストアンサー

numpy.where() をお使いください。

numpy.where — NumPy v1.17 Manual

np.where(条件, 条件式が成り立つ場合の値, 条件式が成り立たない場合の値)

今回10または50の値かどうかで色を分けたいとのことなので、条件式は (img == 10) | (img == 50) になります。

python
1import matplotlib.pyplot as plt
2import numpy as np
3
4# 適当な配列作成
5img = np.random.choice([0, 10, 20, 30, 40, 50], size=(100, 100))
6
7# 10, 50 は白 (255)、それ以外は0
8dst = np.where((img == 10) | (img == 50), 255, 0)
9
10fig, ax = plt.subplots(facecolor="w")
11ax.imshow(dst)
12
13plt.show()