Numpyの範囲外インデックスの折り返し

Question

質問

PythonのNumpyを使ったときの、範囲外のインデックスにアクセスした時に、折り返したインデックスの値を取得／設定したいです。
そのような動作をする簡単な方法をご存知の方がいれば教えてください！

コード例

a = np.arange(10)
a
# -> array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

# 通常であれば範囲外アクセスはエラーとなります。
a[12]
# -> IndexError: index 12 is out of bounds for axis 0 with size 10

# このように折り返したインデックスで値を取得したいです。
a[12]
# -> 2

# 同様に、値の設定も折り返したいです。
a[12] = 12
a
# -> array([ 0,  1, 12,  3,  4,  5,  6,  7,  8,  9])


# 同様に、sliceのアクセスも行いたいです。
a = np.arange(10)
a[8:12]
# -> array([8, 9, 0, 1])

a[8:12] = -1
a
# -> array([-1, -1,  2,  3,  4,  5,  6,  7, -1, -1])

Answer 1

値の取得は numpy.take, 値の設定は numpy.put を使い、mode="wrap" を指定すると質問に記載の内容が実現できます。

numpy.take — NumPy v1.17 Manual
numpy.put — NumPy v1.17 Manual

import numpy as np

a = np.arange(10)
# -> array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

indices = np.arange(8, 12)
print(np.take(a, indices, mode="wrap"))
# [8 9 0 1]

indices = np.arange(8, 12)
np.put(a, indices, -1, mode="wrap")
print(a)
# [-1 -1  2  3  4  5  6  7 -1 -1]

Comments

[tas0g]

回答ありがとうございます！
まさに、私が実現したかったことでした。

Answer 2

インデックスを書くとき n ではなく n % 10 と書けば良いでしょう。
あるいは勝手に折り返してくれる方法を探しているのでしょうか？

Comments

[tas0g]

回答ありがとうございます！
勝手に折り返してくれる方法を探しています。
特に、sliceでアクセスするときに、勝手に折り返してくれる方法があればと思って質問しました。

[LouiS0616]

あ、この方法だとスライスの方には対応していませんね。
とりあえず範囲外アクセスを回避する方法ということで。

[LouiS0616]

自動的に折り返したいなら、適当にラッパークラスを組むことになると思います。
私の知る範囲ではNumPyにそのような機能は無いです (私が知らないだけかもしれませんが) 。

[tas0g]

回答ありがとうございます。
NumPyそのような機能がなさそうだということを確認できたので、助かりました。

[LouiS0616]

tiitoiさんの方法で実現できますね。不勉強でした。