数组迭代

82 阅读1分钟

数组迭代

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.]

数组操作

更改形状