NumPy配列でn番目からm番目までのスライスを取りたいです。nは正の時もあり、負の時もあります。

Question

前提・実現したいこと

NumPy配列でn番目からm番目までのスライスを取りたいです。nは正の時もあり、負の時もあります。
すごく初歩的な質問で申し訳ありません。

該当のソースコード

n= -1
m= 3

a = [1, 2, 3, 4, 5]
b = a[n: m]
print(b)

[]

となり結果が出ません

n = -1
m = 3

a = [1, 2, 3, 4, 5]
a_len = len(a)
a[a_len:a_len] = a
b = a[n + a_len : m + a_len]
print(b)

とすると

[5, 1, 2, 3]

と、求める出力になり、nは正負関係なく使えますが、スマートではありません。
もっと分かりすく使える手段はないでしょうか。

Answer 1

takeを使うのはどうでしょう。

import numpy as np

n = -1
m = 3

a = np.array([1, 2, 3, 4, 5])
b = a.take(np.arange(n, m))

print(b) #[5, 1, 2, 3]

ozwkさんのと組み合わせると、mが大きくなっても大丈夫です。

L = len(a)
b = a.take(np.arange(n, m) % L)

Comments

[ysk_snn]

takeを使ったことがなかったので、勉強になりました！
ありがとうございます！

Answer 2

n = -1
m = 3
a = np.array([1, 2, 3, 4, 5])

a[np.arange(n, m) % len(a)]
# array([5, 1, 2, 3])

Answer 3

n < mという前提のコードです。もし n >= m の場合を考えないといけないなら場合分けが要りそうです。

n= -1
m= 3

a = [1, 2, 3, 4, 5]
b = [a[i] for i in range(n,m)]

print(b)  # -> [5, 1, 2, 3]

Answer 4

考えてみました。

from operator import itemgetter
n = -1
m = 3

a = [1, 2, 3, 4, 5]
b = itemgetter(*range(n,m))(a)

print(b) #[5, 1, 2, 3]

あ、「NumPy配列」という前提を無視しちゃってますが...

Answer 5

うーむ...

n= -1
m= 3

a = [1, 2, 3, 4, 5]
L = len(a)
b = [a[i%L] for i in range(n,m)]

print(b) #[5, 1, 2, 3]

Comments

[ysk_snn]

マイナスの割り算の余りを使う発想すごいです！！
スマートですね！

皆さんのご回答もみてみたいので、もうちょっと待ってみます！