python - 从矩阵创建随机数组

Question

我有以下代码用于从矩阵返回数组，但它不起作用。我想知道你是否可以帮我改正它。非常感谢。

import numpy as np

class city:
    def  __init__(self,A,route):
        self.A=A
        self.route=route

    def  distance(self):
        A = np.array([[ 0,  10,    20,  30],[10,   0,    25,  20],[20,  25,     0,  15],[30,  20,    15,   0]])
        return A

    def route(self,A):
        route = random.sample(A, len(A[:,0]))
        return route

ob=city(route)
print(ob.route)                        

预期输出:

[(0,1),(1,2),(2,3)]

Answer 1

将 numpy 导入为 np随机导入类(class)城市: def init(自身): self.distance()

def  distance(self):
    self.A = np.array([[ 0,  10,    20,  30],[10,   0,    25,  20],[20,  25,     0,  15],[30,  20,    15,   0]])
    self.B = [[random.randint(0,1) for j in range(self.A.shape[0])] for i in range(self.A.shape[1])]

def route(self):
    n=self.B.shape[0]
    nodes=list(range(n))
    route = [nodes.pop(random.sample(range(n-i),1)[0]) for i in range(n)]
    return [(route[i],route[i+1]) for i in range(n-1)]

ob=城市()打印(ob.route())