运行程序直到 networkx.check_planarity() 返回 True

Question

我有以下 function 生成图表。

import numpy as np
import networkx as nx
import random
def random_graph(vertices, connectivity):
    #Creates random symmetric graph
    arr = np.random.randint(0,10,(vertices,vertices))
    sym = (arr+arr.T)
    
    #removing self loops with fixing diagonal
    np.fill_diagonal(sym,0)
    
    #connectivity of graph -> 0 for highest connections, 9 for least connections
    mat = (sym>connectivity).astype(int)
    
    #convert to dictionary
    G = {k:[i for i,j in enumerate(v) if j==1] for k,v in enumerate(mat)}
    return G

我使用带有随机生成连接的 function，不幸的是生成的图并不总是平面的，所以我想运行 function 直到我得到一个平面图。 使用来自networkx的 function check_planarity检查图形的平面性。 function check_planarity返回一个 boolean，如果为真，则该图是平面的，如果为假，则该图不是平面的，我需要生成一个新图，直到它是平面的。 这是我尝试过的：

while True:
    random_connectivity = random.randint(4, 9)
    G = random_graph(5, random_connectivity)
    g = nx.Graph(G)
    planar = nx.check_planarity(g)
    if planar:
        break
    else: 
        pass

正如您从上面的循环中看到的那样，首先我生成random_connectivity ，然后我创建具有该连接性的图形，然后使用networkx检查平面性。 然后我尝试检查它是否是平面的，如果不是，我再次运行程序。 不幸的是，它不起作用，因为它经常返回非平面图。 我还尝试解压planar并检查解压列表的第一个元素，但它不起作用：

planar_check = [x[0] for x in planar]

尝试上面的代码行时，我得到以下回溯：

Traceback (most recent call last):
  File "c:\Users\besan\OneDrive\Desktop\LAM da correggere\test.py", line 64, in <module>  
    planar_check = [x[0] for x in planar]
  File "c:\Users\besan\OneDrive\Desktop\LAM da correggere\test.py", line 64, in <listcomp>
    planar_check = [x[0] for x in planar]
TypeError: 'bool' object is not subscriptable

Answer 1

通过直接检查planar insead 的第一个元素来解决它：

planar = nx.check_planarity(g)[0]