マスキング:指定した条件に従って配列の要素を簡単に抽出できる

python

Numpyのマスキング機能は、指定した条件に従って配列の要素を簡単に抽出するための機能です。この機能を使用すると、配列の要素を条件に基づいて抽出し、新しい配列を作成することができます。

例えば、以下のような配列があるとします。

import numpy as np
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])

この配列から、値が5以上の要素を抽出したい場合、以下のようにマスキングを使用します。

mask = arr >= 5
result = arr[mask]
print(result)
# [5, 6, 7, 8, 9]

この例では、arr >= 5という条件を指定して、マスキングを行いました。その結果、arr[mask]という新しい配列が作成され、値が5以上の要素が抽出されました。

タイトルとURLをコピーしました