利用Python编写一个函数,用于显示nXn的矩阵。其中n是用户输入,矩阵中的值为1或0,由计算机随机产生。矩阵形式如下。
时间: 2024-12-22 16:17:55 浏览: 3
当然,你可以使用Python的`numpy`库来创建一个随机的二维数组(矩阵),然后用`print`函数显示出来。以下是一个简单的函数,它接受一个整数参数`n`,生成一个`n x n`的二进制矩阵:
```python
import numpy as np
def display_random_matrix(n):
# 创建一个n x n的全零矩阵
matrix = np.random.randint(2, size=(n, n))
# 打印矩阵
print("Matrix:")
for row in matrix:
print(row)
# 用户输入n的值
n = int(input("Enter the size of the square matrix (n): "))
display_random_matrix(n)
```
这个函数首先生成一个`n x n`的随机矩阵,其中每个元素都是0或1,这是通过`np.random.randint(2)`得到的。然后,它遍历并打印每行数据。
如果你想要用户自己选择是否生成1或0,可以稍微修改一下:
```python
def generate_binary_choice():
choice = input("Do you want to use only 1s or 0s? Enter '1' for ones and '0' for zeros: ")
if choice == '1':
return 1
elif choice == '0':
return 0
else:
raise ValueError("Invalid choice. Please enter '1' or '0'.")
# 生成随机矩阵时使用用户的选择
matrix = np.random.choice([generate_binary_choice()], size=(n, n))
# ...
```
现在,当你运行程序时,它会询问用户是否要生成1或0的矩阵。
阅读全文