初心者のため,質問の仕方が判りにくかったら申し訳ありません.

Python3で行列を作成し,inputの結果によって行列内の要素を置換するコードを書こうとしています.

図中のbasic_matrixで任意の要素数の行列を作成するまでは出来たのですが,行列内の要素全てについてinputで「"関係性がある(1)か,ない(0)か"」を入力させ,その値をbasic_matrixに反映させるためには,どのようにコードを書けば良いでしょうか.
とりあえず今は画像のように,入力値に応じて文書を返すようにしたのですが,それもうまくいっていません.

画像の説明をここに入力