写一段电网的代码

电网的代码如下：

class PowerGrid: def init(self, num_of_nodes): self.num_of_nodes = num_of_nodes self.grid = [[0 for x in range(num_of_nodes)] for y in range(num_of_nodes)]

def add_edge(self, i, j):
    self.grid[i][j] = 1
    self.grid[j][i] = 1

def remove_edge(self, i, j):
    self.grid[i][j] = 0
    self.grid[j][i] = 0

def get_neighbors(self, node):
    neighbors = []
    for i in range(self.num_of_nodes):
        if self.grid[node][i] == 1:
            neighbors.append(i)
    return neighbor