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]