どうも、ちょげ(@chogetarou)です。
Numpy配列にフィルターをかける方法を紹介します。
方法

Numpy配列にフィルターをかけるには、[]を使います。
まず、Numpy配列名と[]を記述します。
そして、[]内に、Numpy配列と比較する条件式を指定します。
#arr=対象のNumpy配列, condition=Numpy配列と比較する条件式
result = arr[condition]
上記の[]は、[]内に指定した条件式を満たすNumpy配列の要素を全て取得します。
使用例
import numpy as np
nums = np.array([1, 2, 3, 4, 5, 6])
result = nums[nums % 2 == 0]
print(result)
出力:
[2 4 6]
コメント