如何使用NumPy(Python)截断矩阵

pip install numpy

import numpy as np

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

# 设置截断阈值
threshold = 5

# 截断矩阵
truncated_matrix = np.where(matrix< threshold, matrix, threshold)

print(truncated_matrix)

[[1 2 3]
 [4 5 6]
 [5 5 5]]