python在5000*4000二维矩阵中求每个20*20二维移动窗口内极大值的坐标
时间: 2023-04-02 20:04:55 浏览: 111
Python获取二维矩阵每列最大值的方法
5星 · 资源好评率100%
可以使用numpy库中的stride_tricks函数来实现这个功能。具体步骤如下:
1. 将二维矩阵转换为一维数组,方便后续操作。
2. 使用stride_tricks函数将一维数组转换为二维数组,每个子数组的大小为20*20。
3. 对每个子数组求最大值的坐标。
4. 将每个子数组的最大值坐标转换为在原始二维矩阵中的坐标。
代码示例:
import numpy as np
# 生成一个500*400的随机二维矩阵
matrix = np.random.rand(500, 400)
# 定义移动窗口大小
window_size = 20
# 将二维矩阵转换为一维数组
flat_matrix = matrix.ravel()
# 使用stride_tricks函数将一维数组转换为二维数组
shape = (matrix.shape[]-window_size+1, matrix.shape[1]-window_size+1, window_size, window_size)
strides = (matrix.strides[], matrix.strides[1], matrix.strides[], matrix.strides[1])
sub_matrices = np.lib.stride_tricks.as_strided(flat_matrix, shape=shape, strides=strides)
# 对每个子数组求最大值的坐标
max_indices = np.unravel_index(np.argmax(sub_matrices, axis=(2,3)), shape[:2])
# 将每个子数组的最大值坐标转换为在原始二维矩阵中的坐标
max_indices = (max_indices[]+window_size//2, max_indices[1]+window_size//2)
print(max_indices)
阅读全文