python 沿角度而非轴求numpy数组中的元素

Question

是否可以沿角度而非轴对 numpy 数组中的元素求和。

我正在使用二维数组，只能沿 axis=0 或 axis=1 求和。
我想要的是总结例如45或60度。

示例:
矩阵:[[1, 2], [3, 4]] 和角度:45 度。
结果应类似于 [3, 1+4, 2] = [3, 5, 2] (从左上角到右下角求和)。

有人有想法吗？

Answer 1

对于你所说的“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]