networkxを利用したグラフで条件に合った頂点を削除する方法

Question

###実現したいこと
プログラム内で設定したグラフにおいて、「次数が0または次数が1の頂点とその頂点に接続する辺」を削除したい。networkxを使用します。
###問題点
以下のようなエラーがでます。

Traceback (most recent call last):
  File "kadai06d.py", line 10, in <module>
    for v in G:
RuntimeError: dictionary changed size during iteration

上のエラーを見るとfor v in G:の部分にエラーが出ていますが私が考えるに悪い部分はG_deg=G.remove_node(v)だと思います。これを消したらプログラムが正常に機能したからです。
for内で特定の頂点を消すにはどのようにしたらよいでしょうか。教えてください。
###コード

import networkx as nx
import matplotlib.pyplot as plt
G=nx.Graph()
G.add_nodes_from([1,2,3,4,5,6,7,8,9,10])
G.add_edges_from([(1,2),(2,4),(3,2),(4,5),(6,10),(7,6),(8,6),(9,8),(10,8),(1,9),(8,1)])

#n=G.number_of_nodes()
#[次数が0の頂点]もしくは[次数が1の頂点とその頂点に接続する辺]を削除する
for v in G:
    #print(G.degree(v))
    #print("v=",v)
    G_deg=G.degree(v)
    if G_deg==0 or G_deg==1:
        G.remove_node(v)
        #print("G_deg=",G_deg)


#グラフの出力
nx.draw_networkx(G)
plt.show()

Answer 1

「iterate しているオブジェクトを変更することはできない」というエラーです。

先に削除する頂点をリストアップして、Graph.remove_nodes_from() でまとめて削除しましょう。

import networkx as nx

G = nx.Graph()
G.add_nodes_from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
G.add_edges_from(
    [
        (1, 2),
        (2, 4),
        (3, 2),
        (4, 5),
        (6, 10),
        (7, 6),
        (8, 6),
        (9, 8),
        (10, 8),
        (1, 9),
        (8, 1),
    ]
)

nodes_to_remove = [v for v in G if G.degree(v) <= 1]
G.remove_nodes_from(nodes_to_remove)

↓

Comments

[langhtorn]

わざわざグラフつきでありがとうございます。やるべきことがよくわかりました。