list配列から特定の列番号の要素を抜き取って

Question

A=[11,12,13,14,15,16,17,18,19,20]
から特定の列:1,3,6,7列目の要素を抽出して、
B=[12,14,17,18]
を算出したいです。
for文を回して1個1個抽出しても良いのですが、記載が長いです。

matlabの場合は、抽出したい配列数を
C=[1,3,6,7]と定義して、
B=A[C]
とすれば、目的の値をシンプルなコードで得られますが、
pythonで同様記載はできないでしょうか。

Answer 1

numpy 配列にすれば、fancy indexing という機能で指定した添字の値だけを抽出できます。

import numpy as np

A = np.array([11, 12, 13, 14, 15, 16, 17, 18, 19, 20])
B = [1, 3, 6, 7]

print(A[B])  # [12 14 17 18]

# 他にいくつかサンプル例
print(A[[0, 0, 1, 2]])  # [11 11 12 13]
print(A[A % 2 == 0])  # [12 14 16 18 20]
print(A[A > 15])  # [16 17 18 19 20]

Comments

[asaliquid1011]

回答ありがとうございます。
できました。