Python：エッジの自動生成をしたい

Question

前提・実現したいこと

Pythonを使用しています。

• 実現したい内容 …

　「https://programgenjin.hatenablog.com/entry/2019/02/26/075121」にある下記のコードを参考にしています。state**2分のエッジ数を自動生成できるようにしたいです。

#edges = [[0, 0], [0, 1], [1, 0], [1, 2], [2, 0]]
edges = [[0, 0], [0, 1], [1, 0], [1, 1]]

実行したコード

このようなコードを実行しました。
途中まではできたのですが、「???」から先が思いつきません。

In[]:
import numpy as np

state = 2
# zero  = np.zeros((len(state)**2, len(state)))
zero  = np.zeros((state**2, 2))
# for x, y in itertools.product(range(len(state)**2), range(len(state))):
for x, y in itertools.product(range(state**2), range(state)):
    zero[x][y] = ???

得たい結果

望んでいる結果は以下の通りです。

Out[]:
# state=2の時
zero = [[0,0], [0,1], [1,0], [1,1]]

# state=3の時
zero = [[0,0], [0,1], [0,2], [1,0], [1,1], [1,2], [2,0], [2,1], [2,2]]

Answer 1

ごり押しになってしまいましたが、下記のコードを書くことでstate**2分のエッジ数を自動生成できるようになりました。

In[]:
state = 2
edges = np.array([np.sort(np.array([np.arange(state)]*state).flatten()),\
                  np.array([np.arange(state)]*state).flatten()]).T
edges


Out[]:
array([[0, 0],
       [0, 1],
       [1, 0],
       [1, 1]])

Answer 2

ランダムなエッジを自動生成したいのかなと勝手に解釈しました。

state = 2
zero  = np.zeros((len(state)**2, len(state)))

だと、4×2の行列だけど良いんですか？
5×2でやってみました。

import random
import itertools
import numpy as np

row = 5
col = 2
edge  = np.zeros((row, col))
for x, y in itertools.product(range(row), range(col)):
    edge[x][y] = random.randint(0,2)
print(edge)

edge_labels = [0]*row
for i in range(row):
    edge_labels[i] = round(random.random(),1)
print(edge_labels)

Comments

[TeRa_130]

ご回答ありがとうございます。
質問時に私自身が勘違いしていたため、質問内容を精査し編集しました。

Answer 3

import itertools

state = 2
zero  = []
for x, y in itertools.product(range(state+1), range(state)):
    if [x, y] == [1, 1]:
        pass
    else:
        zero.append([x, y])
print(zero)
#[[0, 0], [0, 1], [1, 0], [2, 0], [2, 1]]