NumPy

[Python] NumPy – 2次元配列から条件に合う要素を編集する方法

2025-01-18更新日: 2025-01-18

NumPyでは、2次元配列から条件に合う要素を編集するために、ブールインデックスを使用します。

ブールインデックスは、条件に基づいてTrueまたはFalseの配列を生成し、それを使って特定の要素を選択・編集できます。

例えば、arr[arr > 5] = 0のように記述すると、配列arrの中で5より大きい要素を0に置き換えることができます。

条件は複数組み合わせることも可能で、&や|を使って論理演算を行います。

目次から探す

条件に基づく要素の選択
条件に合う要素の編集方法
応用例：特定の条件に基づく操作
まとめ

条件に基づく要素の選択

ブールインデックスの基本

NumPyでは、ブールインデックスを使用して配列の要素を選択できます。

ブールインデックスとは、条件を満たす要素に対してTrue、満たさない要素に対してFalseを返す配列を作成し、その配列を使って元の配列から要素を抽出する方法です。

以下はその基本的な使い方です。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 条件に基づくブールインデックスの作成
boolean_index = array_2d > 5
# ブールインデックスを使って要素を選択
selected_elements = array_2d[boolean_index]
print(selected_elements)

[6 7 8 9]

この例では、配列の中で5より大きい要素を選択しています。

条件式を使った要素の選択

条件式を使うことで、特定の条件に合致する要素を簡単に選択できます。

例えば、配列内の偶数を選択する場合、次のように記述します。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 偶数を選択する条件式
even_elements = array_2d[array_2d % 2 == 0]
print(even_elements)

[2 4 6 8]

このコードでは、配列内の偶数を選択しています。

条件式を使うことで、柔軟に要素を選択できます。

複数条件の組み合わせ(&と|の使用)

複数の条件を組み合わせて要素を選択することも可能です。

&(AND)や|(OR)を使って条件を組み合わせます。

以下はその例です。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 複数条件を使った要素の選択
selected_elements = array_2d[(array_2d > 3) & (array_2d < 8)]
print(selected_elements)

[4 5 6 7]

この例では、4より大きく8より小さい要素を選択しています。

複数の条件を組み合わせることで、より具体的な要素の選択が可能です。

np.whereを使った条件に基づく要素の選択

np.where関数を使用すると、条件に基づいて要素を選択することができます。

この関数は、条件を満たす要素のインデックスを返します。

以下はその使い方です。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# np.whereを使った条件に基づく要素の選択
indices = np.where(array_2d > 5)
selected_elements = array_2d[indices]
print(selected_elements)

[6 7 8 9]

この例では、np.whereを使って5より大きい要素のインデックスを取得し、それに基づいて要素を選択しています。

np.whereは条件に合う要素の位置を簡単に取得できる便利な関数です。

条件に合う要素の編集方法

ブールインデックスを使った要素の編集

ブールインデックスを使用すると、条件に合う要素を直接編集することができます。

以下の例では、配列内の特定の条件を満たす要素を新しい値に変更します。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 5より大きい要素を10に変更
array_2d[array_2d > 5] = 10
print(array_2d)

[[ 1  2  3]
 [ 4  5 10]
 [10 10 10]]

この例では、5より大きい要素をすべて10に変更しています。

ブールインデックスを使うことで、条件に合う要素を簡単に編集できます。

np.whereを使った要素の編集

np.whereを使うことで、条件に基づいて要素を編集することも可能です。

以下の例では、条件に合う要素を新しい値に置き換えています。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# np.whereを使って条件に合う要素を変更
array_2d = np.where(array_2d < 5, 0, array_2d)
print(array_2d)

[[0 0 0]
 [0 5 6]
 [7 8 9]]

この例では、4未満の要素を0に変更しています。

np.whereを使うことで、条件に基づいた柔軟な編集が可能です。

条件に合う要素を特定して値を変更する

条件に合う要素を特定し、その値を変更する方法もあります。

以下の例では、特定の条件を満たす要素を新しい値に変更しています。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 6以上の要素を2倍に変更
array_2d[array_2d >= 6] *= 2
print(array_2d)

[[ 1  2  3]
 [ 4  5 12]
 [14 16 18]]

この例では、6以上の要素を2倍に変更しています。

条件に合う要素を特定して値を変更することで、より具体的な操作が可能です。

複数条件を使った要素の編集

複数の条件を組み合わせて要素を編集することもできます。

以下の例では、複数の条件を使って要素を変更しています。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 4未満の要素を-1、8以上の要素を99に変更
array_2d[(array_2d < 4)] = -1
array_2d[(array_2d >= 8)] = 99
print(array_2d)

[[-1 -1 -1]
 [ 4  5  6]
 [ 7 99 99]]

この例では、4未満の要素を-1に、8以上の要素を99に変更しています。

複数条件を使うことで、より複雑な編集が可能になります。

応用例：特定の条件に基づく操作

条件に合う要素を別の値に置き換える

条件に合う要素を別の値に置き換える操作は、データの前処理やクリーニングにおいて非常に重要です。

以下の例では、配列内の特定の条件を満たす要素を新しい値に置き換えています。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 5より小さい要素を-1に置き換える
array_2d[array_2d < 5] = -1
print(array_2d)

[[-1 -1 -1]
 [-1  5  6]
 [ 7  8  9]]

この例では、5未満の要素を-1に置き換えています。

条件に基づいて値を変更することで、データの整形が可能です。

条件に合う要素を削除する方法

条件に合う要素を削除することもできます。

NumPyでは、条件に合う要素を除外した新しい配列を作成することができます。

以下の例を見てみましょう。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 5より小さい要素を削除
filtered_array = array_2d[array_2d >= 5]
print(filtered_array)

[5 6 7 8 9]

この例では、5未満の要素を削除し、新しい配列を作成しています。

条件に合う要素を削除することで、必要なデータだけを抽出できます。

条件に合う要素を別の配列に抽出する

条件に合う要素を別の配列に抽出することも可能です。

以下の例では、条件に合う要素を新しい配列に格納しています。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 6以上の要素を別の配列に抽出
extracted_elements = array_2d[array_2d >= 6]
print(extracted_elements)

[6 7 8 9]

この例では、6以上の要素を新しい配列に抽出しています。

条件に合う要素を別の配列に格納することで、データの分析や処理が容易になります。

条件に合う要素の集計や統計処理

条件に合う要素に対して集計や統計処理を行うこともできます。

以下の例では、条件に合う要素の合計や平均を計算しています。

import numpy as np
# 2次元配列の作成
array_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 5以上の要素の合計と平均を計算
filtered_elements = array_2d[array_2d >= 5]
total = np.sum(filtered_elements)
mean = np.mean(filtered_elements)
print(f"合計: {total}, 平均: {mean}")