Python - 将周围元素转换为 K

在处理图像或矩阵时，经常需要将周围的元素转换为某个值，这个操作在Python中非常容易实现。本文将介绍使用Python实现将周围元素转换为K的方法。

方法一：使用Numpy

使用Numpy中的pad函数可以很方便地对数组进行边界填充。然后再将对应位置的元素更改为K即可。

import numpy as np

def convert(arr, k):
    pad_arr = np.pad(arr, (1,1), mode='constant', constant_values=0)
    for i in range(1, pad_arr.shape[0]-1):
        for j in range(1, pad_arr.shape[1]-1):
            if pad_arr[i][j] != k:
                pad_arr[i-1:i+2, j-1:j+2] = k
    return pad_arr[1:-1, 1:-1]

方法二：使用纯Python

当然，如果不想依赖第三方库，我们也可以使用纯Python来实现。具体的实现方法类似于方法一，只是需要手动对数组进行边界填充。

def convert(arr, k):
    h, w = len(arr), len(arr[0])
    pad_arr = [[0] * (w+2)]
    pad_arr += [[0] + row + [0] for row in arr]
    pad_arr += [[0] * (w+2)]
    for i in range(1, h+1):
        for j in range(1, w+1):
            if pad_arr[i][j] != k:
                pad_arr[i-1][j-1:j+2] = [k] * 3
                pad_arr[i][j-1:j+2] = [k, k, k]
                pad_arr[i+1][j-1:j+2] = [k] * 3
    return [row[1:-1] for row in pad_arr[1:-1]]

以上就是使用Python将周围元素转换为K的两种方法。