遍历二维数组,查看相邻数组(边界除外)

2024-06-25 07:24:18 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一段代码,我试图遍历一个dims300350的2d(numpy)数组。 我观察数组中的每个点[I,j],然后观察它的两个相邻点[I+1,j+1]等等(取决于角度)。我看的是0度,45度,90度和135度角。我的问题是索引,对于第一个i,j(角度=0度),我要看i,j+1和i,j-1。 因为第一次迭代是在j=0,我得到一个索引错误,因为j-1=-1。我已经试了好几个小时,研究如何规避这个问题。我的代码如下所示:

for i in range(trex.shape[0]): # ALL ROWS
    for j in range(trex.shape[1]): # ALL COLUMNS
        if shape[i,j] == 0:    # for 0 degree angles (shape is angles)
            if trex[i,j] > trex[i, j+1] and trex[i,j] > trex[i,j-1]: # conditions
                np.append(trex_other, trex[i,j])
            else:
                np.append(trex_other, 0)

我看了所有的角度,但这只是一个例子,我希望它做什么。总而言之,我想在除了二维数组周围的边界之外的所有点上遍历这个二维数组。 非常感谢您的帮助!你知道吗


Tags: 代码inforifnprange数组all