np二维矩阵元素如果不是np.inf就赋值为1

您可以使用以下代码将二维矩阵中非inf元素赋值为1：

import numpy as np

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

# 将非np.inf元素赋值为1
matrix[matrix != np.inf] = 1

# 打印结果
print(matrix)

输出结果：

array([[1., 1., inf],
       [1., inf, 1.],
       [1., 1., 1.]])

可以看到，非np.inf的元素都被赋值为1了。