2024-04-27 05:19:50 发布
网友
我有一个2D数组:arr = np.array([[1,2,3],[1,2,3],[1,2,3]])我想把这些列乘以一列,得到[1,8,27]。是否有numpy函数来执行此操作?我可以通过prod = reduce(lambda x,y:x*y,arr)来完成,但感觉好像我错过了一些明显的numpy函数。。。。你知道吗
arr = np.array([[1,2,3],[1,2,3],[1,2,3]])
[1,8,27]
prod = reduce(lambda x,y:x*y,arr)
In [64]: arr.prod(axis=0) Out[64]: array([ 1, 8, 27])
相关问题 更多 >
编程相关推荐