numpyのwhere関数の使い方は何ですか。
numpyのwhere関数は、与えられた条件に基づいて元の配列から要素を選択するために使用されます。使用法は次のとおりです:
numpy.where(条件[, x, y])
条件:
- 条件:満たす必要がある条件、ブール配列またはブール式のいずれかである。
- xとy:オプションパラメータであり、条件を満たす要素はxの要素に置き換え、条件を満たさない要素はyの要素に置き換えます。xとyは配列またはスカラーであることができます。
戻り値:
- 元の配列と同じ形状の新しい配列を返す。
例:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr > 3, arr, 0)
print(new_arr)
# 输出:[0 0 0 4 5]
上記の例では、arr > 3 はブール配列であり、条件を満たす要素はTrueとなり、条件を満たさない要素はFalseとなります。条件に基づいて、条件を満たす要素をarr内の対応する要素で置き換え、条件を満たさない要素を0で置き換えて新しい配列new_arrが得られました。