沿着角度而不是轴的numpy数组中的python sum元素

Question

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

我使用的是2d数组，只能在axis=0或axis=1之间求和。
我想要的是总结一下学位45或60。

实施例
矩阵：[[1, 2], [3, 4]]和角度：45度。
结果应该是[3, 1+4, 2] = [3, 5, 2]（从左上角到右下角的总和）。

有人有想法吗？

Answer 1

对于你所说的内容很简单＆＃34; 45度＆＃34;：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]