python - 乱数行列で単語を非表示にするにはどうすればよいですか?

Question

私は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

ありがとう！