从2D Numpy阵列中提取对角线交叉轮廓

时间:2016-07-06 15:49:51

标签: python numpy

如果我有一个100x100的2D Numpy阵列,我想以25度的角度(水平方向)提取以(50,50)为中心的交叉轮廓。如何实现这一目标?

我尝试过使用numpy.diagonal无济于事。

1 个答案:

答案 0 :(得分:1)

这就是我要做的。假设你的垫子是mymat。

import numpy as np

ang = 25

extract = []
for i in range(100):
    j = int(50 + (50 - i) * np.tan(ang * np.pi /180))
    if j<=99 and j>=0:
        extract.append(mymat[i,j])

应根据其他尺码调整数字。