📜  python - 检索未连接的节点对 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:13.284000             🧑  作者: Mango

代码示例1
nodelist = list(G0.nodes(data=True))                        # From G to list
vertex = [a for a, b in nodelist]                           # Get attribute vertex
node_df = pd.DataFrame({'vertex': vertex})                  # Get DF with vertex
node_list = node_df["vertex"].tolist()                      # From df to list

node_list = list(dict.fromkeys(node_list))                  # Remove duplicate items from the list
adj_G = nx.to_numpy_matrix(G0, nodelist = node_list)        # Build adjacency matrix


all_unconnected_pairs = []                                                  
offset = 0                                                                  
for i in tqdm(range(adj_G.shape[0])):                                       # Take only above diagonal of the adjacency matrix                                         
  for j in range(offset, adj_G.shape[1]):
    if i != j:
        # if G0[node_list[i], node_list[j]]['weight'] <=2:                  # Take only edges with weight 1 or 2
            if adj_G[i,j] ==0:                                              # Take only unconnected node pairs                                     
                all_unconnected_pairs.append([node_list[i], node_list[j]])

  offset = offset + 1

len(all_unconnected_pairs)