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

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

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

[![画像の説明をここに入力](https://i.stack.imgur.com/PGEIg.jpg)](https://i.stack.imgur.com/PGEIg.jpg)
CC BY-SA 4.0
2019-01-03T15:07:36.750
51683
2019-01-04T06:54:25.790
2019-01-03T16:11:25.667
3605
31657
question
python numpy
for文とif文を使ったリスト内の要素の書き換えについて
inputは文字列（str）を受け取っているので、整数（int）と比較するとすべてNGになります。

```

    if int(input("関係性がある場合…以下略"))==1:
    
```

という風に整数に変換すれば入力値にちゃんと反応するようになります。

もし配列（basic_matrix）の中身と比較したいのであれば、

```

    print(basic_matrix ==1)
    
```

上記をfor文の外に書いてやれば配列の形のまま、合致するものをTrue、そうでないものをFalseで返します。↓

```

    [[ True False False False False]
     [False  True False False False]
     [False False  True False False]
     [False False False  True False]
     [False False False False  True]]
    
```

参考になれば幸いです。
CC BY-SA 4.0
2019-01-04T06:54:25.790
51693
2019-01-04T06:54:25.790
30590
51683
answer