[python] 配列やリストにおけるインデックスの使い方が分からずに困っています

Question

A配列の中にある数字と合致する「B配列の中の数字のインデックス」を返し、
その後、そのインデックスをCリストのインデックスとして使いたいのですが、
以下のようなコードでCリストの情報を取り出そうとすると、エラーが出てしまいます。

どうすれば改善されるのか、どなたかご教示いただけないでしょうか？

A_array=np.array([8551, 8552, 8553, 8554, 8555, 8556, 8557, 8558, 8559, 8560, 8561,
       8562, 8563, 8564, 8565, 8566, 8567, 8568, 8569, 8570, 8571, 8572,
       8573, 8574, 8575, 8576, 8577, 8578, 8579, 8580, 8581, 8582, 8583,
       8584, 8585, 8586, 8587, 8588, 8589, 8590, 8591, 8592, 8593, 8594,
       8595, 8596, 8597, 8598, 8599, 8600, 8601, 8602, 8603, 8604, 8605,
       8606, 8607, 8608, 8609, 8610, 8611, 8613, 8614, 8615, 8616, 8617,
       8618, 8619, 8620, 8621, 8623, 8624, 8625, 8626, 8627, 8628, 8629,
       8630, 8631, 8633, 8634, 8635, 8636, 8637, 8638, 8644, 8645, 8647,
       8648, 8649, 8650, 8651, 8652, 8654, 8656, 8658, 8659, 8660, 8661,
       8662, 8663, 8664, 8665])

B_array=np.array([ 8551,  8552,  8555,  8556,  8557,  8558,  8559,  8560,  8561,
        8562,  8563,  8564,  8566,  8567,  8568,  8569,  8569,  8570,
        8571,  8572,  8572,  8573,  8574,  8575,  8576,  8577,  8578,
        8581,  8582,  8583,  8584,  8585,  8586,  8587,  8588,  8589,
        8590,  8591,  8592,  8593,  8594,  8595,  8596,  8596,  8597,
        8598,  8599,  8600,  8601,  8602,  8603,  8604,  8606,  8607,
        8608,  8609,  8610,  8611,  8612,  8615,  8617,  8618,  8619,
        8633,  8634,  8635,  8636,  8637,  8638,  8639,  8640,  8641,
        8642,  8643,  8644,  8645, 86456,  8647,  8648,  8649,  8650,
        8651,  8652,  8653,  8654,  8655,  8656,  8657,  8658,  8659,
        8660,  8661,  8605,  8612])

C_list=["A","B","C","D","E","F","G","H","I","J","K","L","M","N"]

＜私が書いたコード＞

print(C_list[np.where(B_array==A_array[4])])

私の上記の記述ではA配列の中にある「8554」の数字に一致したB配列のインデックスを返し、それをCリストのインデックスとして用いてCリスト内の「C」の情報が出力されることを期待しています。

＜エラー内容＞

TypeError: list indices must be integers or slices, not tuple

Answer 1

C_listもnumpyアレイにしないと、そのような記法は使えません。

>>> C_array = np.array(C_list)
>>>
>>> print(C_array[np.where(B_array==A_array[4])])
['C']

あるいは、np.whereの戻り値を数値に直すとか。

>>> np.where(B_array == A_array[4])
(array([2], dtype=int64),)
>>> C_list[np.where(B_array == A_array[4])[0][0]]
'C'

はたまた、B_arrayをいったんリストにしてしまうとか。

>>> C_list[B_array.tolist().index(A_array[4])]
'C'

Comments

[mini1988]

丁寧な回答をいただきまして、誠にありがとうございました！おかげさまで解決することができました。