python初学者提问

Question

创建一个3行3列的ndarray数组，数组元素为1，2、3，…、9这9个数，计算输出其所有的元素的和，每行的平均值以及每列的平均值

Answer 1

>>> import numpy as np
>>> y = np.array([*range(1,10)]).reshape(3,3)
>>> y
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
>>> sum(sum(y)) #总和
45
>>> [i/3 for i in sum(y)] #每一列的平均值
[4.0, 5.0, 6.0]
>>> [i/3 for i in sum(y.T)] #每一行的平均值
[2.0, 5.0, 8.0]

Answer 2

用ndarray数组就可以用numpy的各种函数，给个例子

import numpy as np
arr = np.array([*range(1,10)]).reshape(3,3)
total = np.sum(arr)
print('总和：',total)
mean_x = np.mean(arr,axis=1)
print('行平均值:',mean_x)
mean_y = np.mean(arr,axis=0)
print('列平均值:',mean_y)