Statistical Functions

Statistical Functions

import numpy as np
data = np.array([[4,7,2],[8,1,5],[3,9,6]])

# Whole array
np.sum(data)       # 45
np.mean(data)      # 5.0
np.median(data)    # 5.0
np.std(data)       # standard deviation
np.var(data)       # variance
np.min(data)       # 1
np.max(data)       # 9

# Along axis
np.sum(data, axis=0)    # sum each column: [15 17 13]
np.sum(data, axis=1)    # sum each row: [13 14 18]
np.mean(data, axis=0)   # mean of each column

# Index of min/max
np.argmin(data)         # flat index
np.argmax(data)
np.argmin(data, axis=0) # index per column

# Cumulative
np.cumsum(data)
np.cumprod(data)