Python リストの中から指定したインデックス番号の要素を複数同時に取り出す。

Question

Pythonで、リストの中から指定したインデックス番号の要素を複数同時に取り出す。といったことをしたいと思っております。
具体的には

sample_list = [a, b, c, d, e, f, g]

の中から、1番目、5番目、6番目を取り出すといった流れです。
また、取り出したいインデックス番号は下記のようにランダムに近い形でリスト化されています。

wantnumber_list_all = [[0, 1, 3], [1, 5, 6], [1, 3, 4]]

結果として、

result_list_all = [[a, b, d], [b, f, g], [b, d, e]]

のような形で結果を取得したいと思っております。

考えたこととしては、

result_list_all=[]
for wantnumber_list in wantnumber_list_all:
    result_list=[]
    for i, wantnumber in enumerate(wantnumber_list):
        result_list.append(sample_list[i][wantnumber])
    result_list_all.append(result_list)

として可能かとは思うのですが、　forが入れ子になったり、リストが多かったりと、
無駄が多いように感じます。

アドバイス頂きたいです。よろしくお願い致します。

Answer 1

内包表記にしただけですが、ご参考まで。

sample_list = ["a", "b", "c", "d", "e", "f", "g"]
wantnumber_list_all = [[0, 1, 3], [1, 5, 6], [1, 3, 4]]
result_list = [[sample_list[i] for i in lst] for lst in wantnumber_list_all]
print(result_list)
# [['a', 'b', 'd'], ['b', 'f', 'g'], ['b', 'd', 'e']]

Answer 2

内包表記を使うと以下のようになります。

sample_list = np.array(['a', 'b', 'c', 'd', 'e', 'f', 'g'])
wantnumber_list_all = np.array([[0, 1, 3], [1, 5, 6], [1, 3, 4]])
result_list_all = [[sample_list[i] for i in x] for x in wantnumber_list_all]

実行結果

>>> sample_list = np.array(['a', 'b', 'c', 'd', 'e', 'f', 'g'])
>>> wantnumber_list_all = np.array([[0, 1, 3], [1, 5, 6], [1, 3, 4]])
>>> result_list_all = [[sample_list[i] for i in x] for x in wantnumber_list_all]
>>> print(result_list_all)
[['a', 'b', 'd'], ['b', 'f', 'g'], ['b', 'd', 'e']]

Answer 3

numpy を使用しますが、以下のような処理で

import numpy as np

sample_list = np.array(["a", "b", "c", "d", "e", "f", "g"])
wantnumber_list_all = np.array([[0, 1, 3], [1, 5, 6], [1, 3, 4]])

result_list_all = sample_list[wantnumber_list_all]

print(result_list_all)

以下の結果になります。

[['a' 'b' 'd']
 ['b' 'f' 'g']
 ['b' 'd' 'e']]

Comments

[bsdfan]

numpy使うなら内包使わずに
result_list_all = sample_list[wantnumber_list_all]
でできますよ。

[nnkkmto]

確かにそうですね、、、修正しました！ご指摘ありがとうございます。