是否有一种更Pythonic/numPythonic的方法来进行某种嵌套/分层切片,即更漂亮的版本:
_sum = 0
for i in np.arange(n):
_sum += someFunc(A[i,:])
基本上,我想在行上映射 someFunc
(它接受任何形状的数组并返回一个数字),然后对结果求和。
我一直在考虑np.sum(someFunc(A[:,:]))
,但根据我的理解,这只会将someFunc
映射到整个数组。
最佳答案
如果我理解正确,您可以使用如下列表理解:
sum([someFunc(A[i:]) for i in np.arange(n)])
关于Python:分层切片,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41619320/