Sum 3d numpy array [i,j] along i elements

后端 未结 0 1018
再見小時候
再見小時候 2021-02-13 13:45

For a numpy array of shape([N,N,3]),

[[1,2,3], [1,2,3],[1,2,3]],
[[4,5,6],[4,5,6],[4,5,6]],
[[7,8,9],[7,8,9],[7,8,9]]

How can I get a [N,3] with

相关标签:
回答
  • 消灭零回复
提交回复
热议问题