python numpyで等差数列の配列を作成する

python numpyで等差数列の配列を作成する

pythonで、ライブラリnumpyのarangeを使用して、等差数列の配列を作成するサンプルコードを記述してます。pythonのバージョンは3.8.5を使用してます。

環境

  • OS windows10 pro 64bit
  • python 3.8.5

numpyインストール

numpyをインストールされていない方は、pipでインストールしておきます。

pip install numpy

# Successfully installed numpy-1.19.4

arange使い方

arangeを使用すると、等差数列の配列を簡単に作成することが可能です。

import numpy as np

# 0~9まで差が1の等差数列が作成される
arr = np.arange(10)

print(arr)
# [0 1 2 3 4 5 6 7 8 9]

# 開始位置と終了位置(n-1)を指定
arr = np.arange(1,11)

print(arr)
# [ 1  2  3  4  5  6  7  8  9 10]

# 間隔を2に指定
arr = np.arange(1,11,2)

print(arr)
# [1 3 5 7 9]

float型で作成する場合は、以下となります。

import numpy as np

# float型になる
arr = np.arange(10.0)

print(arr)
# [0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]

print(arr.dtype)
# float64

# 0.1から0.2ずつ
arr = np.arange(0.1,1,0.2)

print(arr)
# [0.1 0.3 0.5 0.7 0.9]

多次元配列の場合は「reshape」を使用して作成します。

arr = np.arange(0.1, 0.7, 0.1).reshape(2, 3)

print(arr)
# [[0.1 0.2 0.3]
#  [0.4 0.5 0.6]]