数组迭代
Numpy 提供除for外更为优雅的遍历方法
apply_along_axis(func1d, axis, arr)
-
例子:
x = np.array([[ 11 , 12 , 13 , 14 , 15 ],
[ 16 , 17 , 18 , 19 , 20 ],
[ 21 , 22 , 23 , 24 , 25 ],
[ 26 , 27 , 28 , 29 , 30 ],
[ 31 , 32 , 33 , 34 , 35 ]])
def my_func (x):#自己定义一个计算函数
return (x[ 0 ] + x[- 1 ]) * 0.5
y = np.apply_along_axis(my_func, 0 , x)#axis=1表示横轴,方向从左到右;axis=0表示纵轴,方向从上到下。
print (y) # [21. 22. 23. 24. 25.]
数组操作
更改形状