在Pyhton中创建具有特定条件的矩阵

pip install numpy

import numpy as np

# 创建一个3x3的零矩阵
zero_matrix = np.zeros((3, 3))
print(zero_matrix)

# 创建一个3x3的单位矩阵
identity_matrix = np.eye(3)
print(identity_matrix)

# 创建一个3x3的随机矩阵
random_matrix = np.random.random((3, 3))
print(random_matrix)

# 创建一个3x3的对角矩阵，对角线元素为1,2,3
diagonal_matrix = np.diag([1, 2, 3])
print(diagonal_matrix)

# 创建一个满足特定条件的矩阵，例如大于等于5的元素为1，小于5的元素为0
conditional_matrix = np.where(diagonal_matrix >= 2, 1, 0)
print(conditional_matrix)