TypeError: 'generator' object is not subscriptable

Question

以下の様なコードでTypeError: 'generator' object is not subscriptableというエラーが出ました。

import networkx as nx

all_shortest = nx.all_pairs_dijkstra_path(self.graph)

        for s in range(self.ACTIVE_NODES):
            for d in range(self.ACTIVE_NODES):
                if s != d:
                    next = all_shortest[s][d][1]  #ここでTypeError: 'generator' object is not subscriptable
                    port = self.ports[s][next]
                    routing_nodes[s][d] = next
                    routing_ports[s][d] = port
                else:
                    routing_nodes[s][d] = -1
                    routing_ports[s][d] = -1

ある論文に載っていたプログラムです。その為動くとは思うのですがエラーが出ているという事は何かしら環境に問題があると思います。他の部分でnetworkxのバージョンの違いによるエラーがあったので、これもnx.all_pairs_dijkstra_path(self.graph)の戻り値であるall_shortestに何か問題があるのでは無いかと思います。しかしそもそもこのエラーはどういう事なのでしょうか？

追記ーーーーーーーーーーーーーーーーーー

以下の様にしました。
すると今度は別の場所でValueError: setting an array element with a sequence.となります、、、

import networkx as nx

routing_nodes = np.full([self.ACTIVE_NODES]*2, -1.0, dtype=int)

all_shortest = list(nx.all_pairs_dijkstra_path(self.graph))

        for s in range(self.ACTIVE_NODES):
            for d in range(self.ACTIVE_NODES):
                if s != d:
                    next = all_shortest[s][d][1]
                    port = self.ports[s][next]
                    routing_nodes[s][d] = next  #ここでValueError: setting an array element with a sequence.
                    routing_ports[s][d] = port
                else:
                    routing_nodes[s][d] = -1
                    routing_ports[s][d] = -1

今度はどういう事でしょうか？

Comments

[LouiS0616]

単にall_shortest[0]と書いたときにエラーは出ますか。

[Hayato1201]

next = all_shortest[0] TypeError: 'generator' object is not subscriptableと同じエラーが出ました。。。

Answer 1

値をリストからジェネレータに置き換えることで、効率化を図っているようです。
その一方で、インデックスを指定してランダムアクセスすることができなくなるのです。

単に現状のコードを動かしたいのなら、次のように修正してみてください。

all_shortest = list(nx.all_pairs_dijkstra_path(self.graph))

Comments

[Hayato1201]

回答ありがとうございます！
all_shortest = list(nx.all_pairs_dijkstra_path(self.graph))で行ったところすぐ後に別のエラーが出てしまいました。。。その部分を追記致しました

Answer 2

変数保持のデフォルトの挙動が変わったので、リスト化してみてください。

a=list(range(5))
print(a[0])
a=range(5)
print(a[0]) #error

---

スライスをとって代入してください。

>>> import numpy as np
>>> a = np.zeros((10,10))
>>> a
array([[ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.]])
>>> b = np.ones((2,2))
>>> b
array([[ 1.,  1.],
       [ 1.,  1.]])
>>> a[1:3, 1:3] = b
>>> a
array([[ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  1.,  1.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  1.,  1.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.]])

Comments

[mkgrei]

最後prontになってません？
編集するとprint(a[0])なんだが…

[LouiS0616]

なってます。テキトーに編集するとおおかた直ります。
それこそ 一文字入れて ⇒ それを消す みたいな編集でも大丈夫だったはず。

[Hayato1201]

回答ありがとうございます！変数保持のデフォルトの挙動が変わったのはnetworkxの話ですよね？

[mkgrei]

python2→python3の話です。
networkxが実装を変えたのかどうかはログを見ないと。

[mkgrei]

1空行追加しました、スッキリ。

dbと表示がずれるの気になります。

[mkgrei]

たぶんpython3に移行したとき、ドキュメントのアップデート漏れ。
Exampleが正しくない。
https://networkx.github.io/documentation/networkx-1.10/reference/generated/networkx.algorithms.shortest_paths.weighted.all_pairs_dijkstra_path.html

[Hayato1201]

了解です！ただ、all_shortest = list(nx.all_pairs_dijkstra_path(self.graph))でリスト化するとまたその先でエラーが出てしまいました・・・その点追記しました！

[mkgrei]

listからさらにnp.arrayにキャストしてみてください。

それ以前にやろうとしていることが誤っている可能性があります。

[Hayato1201]

さらにnp.arrayにキャストという事はrouting_nodes[s][d] = np.array(next)この様にするという事ですよね？するとやはり同じ様にValueError: setting an array element with a sequence.となってしまいます。。。何度もすみませんがお分かりになりますでしょうか？

[mkgrei]

routing_nodes[s][d]は数、np.array(next)は配列なのでエラーになります。
np.array(next)と同じ次元だけrouting_nodes[s][d]を取り出す必要がありますが、
それがそもそもやりたいことなのか疑問があります。

[Hayato1201]

結局やりたいことが不明だった為、networkxのバージョンを下げてnx.all_pairs_dijkstra_pathの戻り値を比較した所ディクショナリ型だった為、リスト型ではなくディクショナリ型でキャストしたところ、ここのエラーは無くなりました！