如何在numpy中以3x3矩阵的形式获取特定数字的周围数字？

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 获取特定数字的位置
target_num = 5
row, col = np.where(matrix == target_num)

# 获取周围数字的索引范围
start_row = max(0, row - 1)
end_row = min(row + 2, matrix.shape[0])
start_col = max(0, col - 1)
end_col = min(col + 2, matrix.shape[1])

# 获取周围数字
surrounding_nums = matrix[start_row:end_row, start_col:end_col]

print(surrounding_nums)

[[1 2 3]
 [4 5 6]
 [7 8 9]]