Retrieve position of elements with setting some criteria in numpy

后端 未结 4 868
没有蜡笔的小新
没有蜡笔的小新 2021-01-13 22:24

For the given 2d array of data, how to retrieve the position (index) of 7 and 11 in the bold. Because only they are the elements surrounded by same value in the neighbours

4条回答
  •  孤街浪徒
    2021-01-13 22:36

    I have tested it and the following code works:

    for i in range(1,np.shape(data)[0]-1):
        for j in range(1,np.shape(data)[1]-1):
            if np.all(data[i-1:i+2,j-1:j+2]==data[i,j]):
                print np.array([i,j], dtype=np.int64)
    

提交回复
热议问题