配列 ベクトル 行列 統計

NumPy の arange は最小値と最大値のある等差数列をつくる

最終更新日 2023.02.18

NumPy の arange は一定の範囲における整数を返します。

import numpy as np

a = np.arange(4)
b = np.arange(2, 7)
c = np.arange(3, 11, 2)
d = np.arange(5, 20, 3)

print(a)  # [0 1 2 3]
print(b)  # [2 3 4 5 6]
print(c)  # [3 5 7 9]
print(d)  # [ 5  8 11 14 17]

引数が 3 つの場合、最後の引数はステップを意味します。この関数は等差数列の作成でよく使われます。