Python リスト内包表記で別のリストのインデックスをとる。さらにそれをリストにする。

Question

前提・実現したいこと

Pythonでリスト内包表記のようにして、以下のようなことを実現したいのですが、このコードでは動きません。なにかよい方法はないでしょうか？

In [2]: data = [i for i in range(0,10)]

In [3]: data
Out[3]: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

In [4]: data[i for i in range(0,10,3)] 
#本当はここでdata[0],data[3],data[6],data[9]をとってそれをリストにしたい。
#最終的に欲しいもの[data[0],data[3],data[6],data[9]]
  File "<ipython-input-4-c051e533c8f1>", line 1
    data[i for i in range(0,10,3)]
             ^
SyntaxError: invalid syntax

試したこと, リスト内包表記

上のIn [4]に加えて

In [3]: data
Out[3]: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

In [5]: data[[i for i in range(0,10,3)]]
----------------------------------------------------------------
TypeError                      Traceback (most recent call last)
<ipython-input-7-b3ef10e249c9> in <module>()
----> 1 data[[i for i in range(0,10,3)]]

TypeError: list indices must be integers, not list

も試しました。

なにか便利な良い方法をご存知の方がいらっしゃいましたら教えていただけるとありがたいです。よろしくお願いいたします。

Answer 1

こんな感じで書けます。

>>> data = [i for i in range(0,10)]
>>> data
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>
>>> [data[i] for i in range(0, 10, 3)]
[0, 3, 6, 9]

スライスも使えます。

>>> data[::3]
[0, 3, 6, 9]

---

NumPyアレイならこっちの書き方も出来ますね。

>>> import numpy as np
>>> data = np.arange(10)
>>>
>>> data
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>>
>>> data[[i for i in range(0, 10, 3)]]
array([0, 3, 6, 9])
>>>
>>> data[range(0, 10, 3)]
array([0, 3, 6, 9])

イテラブルを添え字指定できますので。べんり。スライスも使えます。

>>> data[::3]
array([0, 3, 6, 9])

Comments

[fy_2000]

```python
data = [i for i in range(0,10)]
```
のはじめのiのところはiしかだめなのだと思っていました。
とても勉強になりました。
ありがとうございます。

[LouiS0616]

さりげなく追記しましたが、この場面ではスライスを利用するのが最適です。
けっこう面白くて、かつ強力な機能ですから、積極的につかってみてくださいね。