2012/04/24 (Tue) python: subscripting（固定数ごとに要素を抜き出す）

Rでベクトルの一部を取り出したい時には、要素番号を整数か論理値のベクトルで与える。

x <- c("A", "B", "C", "D", "E", "F")

x[ c(1, 5) ]

x[ c(F, F, T, F, T, T) ]

[1] "A" "E"

[1] "C" "E" "F"

 

だから、例えば2つおきとか3つおきとかで要素を取得したい場合は、そういうインデックスを作って代入すれば良い。例えば、第2要素から始まって3つおきに要素を抜き出すには、[2, 5] というベクトルを作って代入するか、[F, T, F, F, T, F] というベクトルを作って代入する。

x <- c("A", "B", "C", "D", "E", "F")

x[ seq(2, length(x), by=3) ]

x[ (1:length(x) %% 3) == 2 ]

 

[1] "B" "E"

[1] "B" "E"

 

pythonで同じ事をやろうとするとエラーが出る。リストはインデックスとして受け付けられない。

x = ["A", "B", "C", "D", "E", "F"]
x[ [2, 5] ]
x[ [False, False, True, False, True, True] ] 

Traceback (most recent call last):

  File "<stdin>", line 1, in <module>

TypeError: list indices must be integers, not list

ただし、先ほどののように固定要素ごとに抜き出したいのなら、サブスクリプティングという方法でできる。x[i::k] という形で、iがスタート地点、kが間隔。

x = ["A", "B", "C", "D", "E", "F"]
x[1::3]
x[0::2]

['B', 'E']

['A', 'C', 'E']