Python 3.x 获取2D numpy数组中最大值的邻居
我有一个10x10形状的2D数组,我需要在2D数组中找到最大值的邻域。我的代码是:Python 3.x 获取2D numpy数组中最大值的邻居,python-3.x,numpy,Python 3.x,Numpy,我有一个10x10形状的2D数组,我需要在2D数组中找到最大值的邻域。我的代码是: import numpy as np array = np.random.randint(-10,10, size=(10,10)) max_index = np.argmax(array) def get_coordinate_i(value): i = 0 if value % 10 >= 1: i = value // 10
import numpy as np
array = np.random.randint(-10,10, size=(10,10))
max_index = np.argmax(array)
def get_coordinate_i(value):
i = 0
if value % 10 >= 1:
i = value // 10
return i
j = max_index - 10*get_coordinate_i(max_index)
i = get_coordinate_i(max_index)
def fun(i, j, array):
arr = np.array([])
if i!=0 and j!=0 and i!=9 and j!=9:
for val in range(-1,2):
for val2 in range(-1,2):
arr = np.append(arr, array[i+val][i+val2])
return np.reshape(arr, (3,3))
尽管如此,它并不是每次都能正常工作。您可以用不同的分类器方法计算i、j和最大值周围的矩阵
import numpy as np
array = np.random.randint(-10,10, size=(10,10))
max_index = np.argmax(array)
j = max_index%10
i = max_index//10
if j==0:
matrix_around_max =array[i-1:i+2,j:j+2]
elif i ==0:
matrix_around_max = array[i:i+2,j-1:j+2]
elif i==0 and j==0:
matrix_around_max = array[i:i+2,j:j+2]
else:
matrix_around_max = array[i-1:i+2,j-1:j+2]
避免循环:
a=np.random.randint(0100,(10,10))
x、 y=np.Unlavel_索引(a.argmax(),a.shape)
框=a[np.max([x-1,0]):np.min([x+1,9])+1,np.max([y-1,0]):np.min([y+1,9])+1]
请注意,这并不能解释max值多次出现的原因,并且argmax()
方法在第一次出现时停止