python - 制作 0 到 nxn 矩阵并将其 reshape 为对

Question

也许我错了，我不知道如何寻找它。

我需要获取配对列表

(0,0)(0,1)(0,2)...(0,999)...(999,999)

将这些值放入神经网络并得到结果 - 只是一个长度为 (999*999) 的向量

然后我可以将其 reshape 为 (999,999) 并获得输出图。

所以，天真的方法是使用

for i in range(0, 999):
    for j in range(0, 999):
        list.add(predict(i,j))

但是不可能使用批量

另一种幼稚的方法是

numbers=[]
for i in range(0, 999):
    for j in range(0, 999):
        numbers.append([i,j])

这可以批量工作。

还有更优雅的解决方案吗？

Answer 1

试试这个:

target_list = np.array([(i,j) for i in range(1000) for j in range(1000)])

此 print(target_list) 将以 numpy 数组格式给出所需的列表:

[[  0   0]
 [  0   1]
 [  0   2]
 ...
 [999 997]
 [999 998]
 [999 999]]