【问题标题】:Python: Hierarchical SlicingPython:分层切片
【发布时间】:2023-04-04 03:45:01
【问题描述】:

是否有更 Pythonic/numpythonic 的方式来进行某种嵌套/分层切片,即更漂亮的版本:

_sum = 0
for i in np.arange(n):
   _sum += someFunc(A[i,:])

基本上我想将someFunc(它采用任何形状的数组并返回一个数字)映射到行上,然后对结果求和。

我一直在考虑np.sum(someFunc(A[:,:])),但据我了解,这只会将someFunc映射到整个数组。

【问题讨论】:

  • sum(np.apply_along_axis(someFunc,1,A)) 虽然不是矢量化选项?
  • someFunc 应该做什么?
  • 在这种情况下,它是在计算每行相等值的数量……但是我在很多情况下都遇到过这个问题,所以我提出了一个更笼统的问题。

标签:
python
numpy
slice
syntactic-sugar