洋書、時々プログラミング

博士課程修了→メーカーという経路を辿っている人の日常

配列の値を置換するのにNumpyは便利だし早い

例えば配列の値を置換したい場合、下記のようにできる。二次元配列などにも適用可能だし、forを使うよりも早い(はず)

import numpy as np
a = np.array(1,2,1,2,3,4,5)
a[a == 5] = 0 #aの中で5の値を0に置換できる

256*256*3の行列とかで、1番目の要素の値に応じて書き換えたい場合も、下記のように書ける

import numpy as np

#aの宣言は省略
a[a[:,:,0] == 1] = [0,0,0] #aの1番目の要素が1だった場合、要素を[0,0,0]にする