Pythonで配列のインデックスを操作する方法

Pythonでは、配列のインデックスを操作するためのいくつかの方法があります。ここでは、PythonのリストとNumPyのndarrayでの基本的なインデックス操作について説明します。

Pythonのリストでのインデックス操作

Pythonのリストでは、index()メソッドを使用して特定の要素のインデックスを取得することができます。以下に例を示します。

list = ["foo", "bar", "baz"]
index = list.index("bar")
print(index)  # Output: 1

このindex()メソッドは、リストの中で最初に見つかった要素のインデックスを返します。要素がリストに存在しない場合、ValueErrorが発生します。

NumPyのndarrayでのインデックス操作

NumPyのndarrayでは、より高度なインデックス操作が可能です。以下に例を示します。

import numpy as np

# 1次元配列
arr = np.array([1, 2, 3, 4, 5])
index = np.where(arr == 2)
print(index)  # Output: (array([1]),)

# 多次元配列
arr_m = np.arange(12).reshape(2, 2, 3)
print(arr_m[0])  # Output: array([[0, 1, 2], [3, 4, 5]])

np.where()関数は、条件に一致する要素のインデックスを返します。また、多次元配列では、各次元に対するインデックスを指定することで特定の要素にアクセスすることができます。

以上がPythonとNumPyでの基本的な配列のインデックス操作方法です。これらの基本を理解することで、より複雑なデータ操作が可能になります。