我是一个Python初学者,我被困在这里。我编写了下面的程序,生成一个随机数矩阵。
def randsq(size):
for i in range(size):
for j in range(size):
print(random.randint(0,9), end = '')
print()
无论如何,我可以添加一个函数,从输入参数中获取一个单词,并在矩阵中对角线隐藏它?所以如果用户输入'rain',输出看起来像这样:
r123
1a45
23i4
989n
谢谢!
修改代码:
import random
def randsq(word):
size = len(word)
for i in range(size):
for j in range(size):
if i == j: # we are at a diagonal-value
print(word[i], end='') # it's the i-th diag -> choose i-th char
else:
print(random.randint(0, 9), end='')
print()
randsq('Rain')
输出:
R757
0a91
02i9
757n
如果你想让用户输入这个单词:
word_to_hide = input('Which word to hide? (+enter) ') # will wait for user-input + enter
randsq(word_to_hide)