python求均值、中位數、眾數的方法
- 2020 年 1 月 8 日
- 筆記
首先需要數據源,這裡隨便寫了一個:
nums = [1,2,3,4]
- 求均值和中位數均可以使用numpy庫的方法:
import numpy as np #均值 np.mean(nums) #中位數 np.median(nums)
- 求眾數方法一:
在numpy中沒有直接的方法,但是也可以這樣實現:
import numpy as np counts = np.bincount(nums) #返回眾數 np.argmax(counts)
其中np.bincount方法返回了一個長度為nums最大值的列表,列表中的每個值代表其索引位數值出現在nums中的次數,例如
返回[2,1,0],代表0在nums中出現2次,而1在nums中出現1次,3在nums中沒有出現。
然後再使用np.argmax就能得到眾數啦。但是,由於索引值是從0開始的,所以這種求眾數的方法只能用在非負數據集。
- 求眾數方法二——直接利用scipy下stats模組【推薦】:
from scipy import stats stats.mode(nums)[0][0]