備忘録：numpyでインデックス用の連番を振る

数値計算ではインプットファイルなどでインデックス用の連番を振ることよくあります。

テクニックのメモです。

配列を繰り返す

配列0～4を3回繰り返したインデックスの作成方法です。

import numpy as np

x = range(5)
print(list(x))
# [0, 1, 2, 3, 4]

i = np.tile(x, 3)
print(i)
# [0 1 2 3 4 0 1 2 3 4 0 1 2 3 4]

配列の要素を繰り返す

配列0～4の要素を各3回繰り返したインデックスの作成方法です。

i = np.repeat(x, 3)   
print(i)
# [0 0 0 1 1 1 2 2 2 3 3 3 4 4 4]