python-数组计算

1、二维数组求和

a = [[1,2],[3,4],[5,6]]
print(sum(sum(i) for i in a))

2、剔除numpy数组中的0值

import numpy as np
array = [1, 1, 0, 3, 4, 5, 0]
a = np.array(array)
b = a[a != 0]
print(b) # [1 1 3 4 5]

3、numpy一维数组，求和、均值、方差、标准差

import numpy as np
array = [1, 1, 0, 3, 4, 5, 0]
a = np.array(array)
#求和（全部元素相加的和）
print("求和:",a.sum())
#求方差
print("方差:",a.var())
#求标准差
print("标准差:",a.std())
#求平均值
print("均值:",a.mean())

4、三维数组，按照Z轴来取值，

print(np.array(a3)[:, 1, 2])

5、三维数组，按照Z轴取最大值，以及最大值的索引

        

chamax2019 = np.max(cha2019, axis=0)

maxindex2018 = np.argmax(cha20192018, axis=0)

chawhere2019 = np.where(chamax2019 >= 1.2, 1, 0)