python-3.x - 在二维 numpy 数组中获取最大值的邻居
问题描述
我有一个形状为 10x10 的二维数组,我需要在二维数组中找到最大值的邻居。我的代码是:
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))
但是,它并非每次都正常工作。
解决方案
避免循环:
a = np.random.randint(0,100, (10,10))
x, y = np.unravel_index(a.argmax(), a.shape)
box = a[np.max([x-1, 0]):np.min([x+1, 9])+1, np.max([y-1, 0]):np.min([y+1, 9])+1]
请注意,这不考虑最大值的多次出现,并且argmax()
方法在第一次出现时停止。
推荐阅读
- python - 如何将特定文件从一个子目录移动到另一个子目录
- android - 仅在实体右侧使用 Stroke 可绘制
- machine-learning - 通过创建自己的标签进行监督学习
- knex.js - 有没有办法在 Knex.js 中将行转换为列?
- hibernate - 配置spring验证和hibernate验证
- c++ - 没有匹配的函数调用,没有已知的转换
- c# - 命令查询分离 - 返回值的异步命令
- swift - iOS中的CommonCrypto对openssl命令行产生不同的结果
- javascript - 如何动态设置关联数组的命名索引?
- ruby-on-rails - Postgresql 多对多大小优化