NumPy の where で配列から条件にあう要素を抽出する - 100 までの自然数から 8 の倍数を抽出する方法
2023.02.18
NumPy の where は配列から条件にあう要素を抽出します。
import numpy as np
numbers = np.array([1, 2, 3, 4])
data = numbers[np.where(2 < numbers)]
print(data) # [3 4]
配列を数とみたてて大小関係の条件を where に入れています。
100 までの自然数から 8 の倍数を抽出する
import numpy as np
numbers = np.arange(1, 101)
data = numbers[np.where(numbers % 8 == 0)]
print(data) # [ 8 16 24 32 40 48 56 64 72 80 88 96]
