The kernel appears to have died. It will restart automatically.と吐いて計算をやめてしまう対処法について。

Question

The kernel appears to have died. It will restart automatically.と吐いて計算をやめてしまう対処法について。

以下リスト内のリストをすべて外す関数を用いて。

def flatten(data):
    for item in data:
        if hasattr(item, '__iter__'):
            for element in flatten(item):
                yield element
        else:
            yield item
rensyu = [1,[1,2],3,[1,[1,2]]]
print(list(flatten(rensyu)))

上記のコードを実行すると以下のような結果が返ってくる。

[1, 1, 2, 3, 1, 1, 2]

このようになる理由は分かっているつもりだったのですが、

def flatten(data):
    for item in data:
        if hasattr(item, '__iter__'):
            for element in flatten(item):
                yield element
        else:
            yield item
rensyu = [1,[1,2],3,[1,[1,2]],'Hello']
print(list(flatten(rensyu)))

rensyuというリストに'Hello'なる文字列を追加したとき
実行結果が以下のように帰ってきてしまいます。

RecursionError: maximum recursion depth exceeded

そこで以下のようなコードを追加することにより解決するというサイトを発見したので、
https://qiita.com/narupo/items/e25ac05a9065c0bd9c03

import sys
sys.setrecursionlimit(10000)

上記のコードを含め、再び以下のコードを実行したところ

def flatten(data):
    for item in data:
        if hasattr(item, '__iter__'):
            for element in flatten(item):
                yield element
        else:
            yield item
rensyu = [1,[1,2],3,[1,[1,2]],'Hello']
print(list(flatten(rensyu)))

以下の結果が返って来てしまいました。

The kernel appears to have died. It will restart automatically.

私としては、以下のような結果が得られればうれしいです。

[1, 1, 2, 3, 1, 1, 2,'Hello']

なぜエラーが発生してしまうか、
また、得られたらうれしい結果はどのようにすれば出すことができるか
教えて頂けると恐縮です。

宜しくお願い致します。

Answer 1

文字列が要素に含まれた場合、文字列の要素も文字列なのでつねにhasattr(item, '__iter__')がTrueになり無限に再帰関数が呼ばれます。
その結果RecursionError: maximum recursion depth exceededが発生します。
無限に再帰してしまうのでsys.setrecursionlimit(10000)にて再帰回数を制御しても効果はありません。

元のコードを生かしたまま修正するなら
if (not isinstance(item,str)) and hasattr(item, '__iter__'):や
if isinstance(item,list): # リストのみの入れ子の場合
とすれば意図した動作になります。

Comments

[yu__]

非常に簡潔に説明していただきありがとうございました。
理解できました！

Answer 2

pythonでは文字列はイテレータにもなります。
なので、文字列'Hello'が'H', 'e', 'l', 'l', 'o'に分解されて、先頭の'H'がさらに分解されて'H'になって、その先頭の'H'が'H'に分解されて、その先頭の'H'が… と永久に続いていくため再帰呼び出しの段数超過でエラーになっているのだと思います。

期待する結果を出すだけであれば以下のようにif文に条件を追加して「イテレータでも文字列は分解しない」ようにすれば良いと思います。

def flatten(data):
    for item in data:
        if hasattr(item, '__iter__') and not isinstance(item, str):
            for element in flatten(item):
                yield element
        else:
            yield item

Comments

[yu__]

先頭の'H'がさらに分解されて'H'になって、その先頭の'H'が'H'に分解されて、その先頭の'H'が… と永久に続いていくため再帰呼び出しの段数超過でエラーになっているのだと思います。
上記の説明が非常に分かりやすく簡単に理解できました！
数値はイテラブルではないが文字列はイテラブルということですね。
非常に分かりやすく説明していただいたので、ベストアンサーにさせていただきました。