Numpy统计计算

求和

import numpy as np

a = np.arange(1,7).reshape((2,3))
c_sum = np.sum(a, axis = 0)        # 沿行方向，求列和
r_sum = np.sum(a, axis = 1)        # 沿列方向，求行和
c_sum1 = a.sum(axis=0)
r_sum1 = a.sum(axis=1)
print(a, "\n")
print(c_sum, "\n")
print(r_sum, "\n")
print(c_sum1, "\n")
print(r_sum1, "\n")

ls = [[1, 2, 3], [4, 5, 6]]
print(np.sum(ls, axis=0))          # np.sum函数直接对非数组对象求和

求均值

a = np.arange(1,7).reshape((2,3))
c_mean = np.mean(a, axis = 0)      # 列平均值
r_mean = np.mean(a, axis = 1)      # 行平均值
c_mean1 = a.mean(axis=0)
r_mean1 = a.mean(axis=1)
print(a, "\n")
print(c_mean, "\n")
print(r_mean, "\n")
print(c_mean1, "\n")
print(r_mean1)

a = np.arange(1,7).reshape((2,3))
r_mean = np.mean(a, axis=0)                            # 算术平均值，不能设置权重
r_average = np.average(a, axis=0, weights=[0.3, 0.7])  # 加权平均值，权重数组的和为1
print(a, "\n", r_mean, "\n", r_average)

求统计差异

a = np.arange(1,7).reshape((2,3))
a_std = np.std(a, axis=0)             # 计算标准差
a_var = np.var(a, axis=0)             # 计算方差
a_ptp = np.ptp(a, axis=0)             # 计算最大值与最小值之差，极差
a_median = np.median(a)               # 计算中位值
print(a_std, a_var, a_ptp, a_median)