python numpyで分散を求める

python numpyで分散を求める

pythonで、ライブラリnumpyのvarを使用して、分散を求めるサンプルコードを記述してます。pythonのバージョンは3.10.0を使用してます。

環境

  • OS windows10 pro 64bit
  • python 3.10.0

numpyインストール

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

pip install numpy

Successfully installed numpy-1.22.3

var使い方

varを使用すると、配列データから分散を求めることが可能です。

import numpy as np

np.var(配列)

以下は、用意した配列から分散を求めるサンプルコードとなります。

import numpy as np

arr = [10, 20, 30, 40, 20, 50]

print( np.var(arr) ) # 180.55555555555557

print( '{0:.2f}'.format( (np.var(arr)) )) # 180.56 

print(arr) # [10 20 30 40 20 50]

numpyを使用しない

numpyを使用せずに、計算する場合は、以下のコードとなります。

arr = [10, 20, 30, 40, 20, 50]

mean = sum(arr)/len(arr)

v = 0

for i in arr:
	v = v + (1/len(arr)) * (i - mean)**2

print( v ) # 180.55555555555554

print( '{0:.2f}'.format( v )) # 180.56