配列 ベクトル 行列 統計

NumPy の where で配列から条件にあう要素を抽出する - 100 までの自然数から 8 の倍数を抽出する方法

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]

NumPy 配列