是否可以沿角度而非轴对 numpy 数组中的元素求和。
我正在使用二维数组,只能沿 axis=0
或 axis=1
求和。
我想要的是总结例如45或60度。
示例:
矩阵:[[1, 2], [3, 4]]
和角度:45 度。
结果应类似于 [3, 1+4, 2] = [3, 5, 2]
(从左上角到右下角求和)。
有人有想法吗?
最佳答案
对于你所说的“45度”来说很容易:numpy trace
import numpy as np
a = np.array([[1,2],[3,4]])
np.trace(a)
5
np.trace(a, offset=1)
2
np.trace(a, offset=-1)
3
并作为列表:
>>> [np.trace(a,offset=i) for i in range(-np.shape(a)[0]+1, np.shape(a)[1])]
[3, 5, 2]
关于python 沿角度而非轴求numpy数组中的元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50700256/