リスト内のリストをすべて外す方法について

Question

リスト内のリストをすべて外す方法について

ある関数を用いて

[[1, 2], [3, 4, 5]]

を

[1, 2, 3, 4, 5]

という変換をしたいと思ったので、
https://qiita.com/shiracamus/items/fb85943ed34d5ec09c4f
というサイトを見ていたのですが
このサイト内の以下の関数の

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

hasattr(item, '__iter__')

上記の部分がわかりません
いろいろなサイトを確認はしましたがいかんせん初心者なので何を行なっているか
わかりませんでした。
もしわかる方がいれば教えていただけると恐縮です。

よろしくお願いいたします。

Answer 1

リストを平滑化する方法

関数flattenは、次のように複雑にネストしたリストでも平滑化できるよう工夫されています。

[1, [2, 3], 4, [5, [6, 7]], 8] ⇒ [1, 2, 3, 4, 5, 6, 7, 8]

ここまで複雑なリストを扱うので無ければ、関数は次のように書けば充分です。

def flatten_2d(data):
    for block in data:
        for elem in block:
            yield elem


lst = [[1, 2], [3, 4, 5]]

dst = list(flatten_2d(lst))
print(dst)

実行結果 Wandbox

[1, 2, 3, 4, 5]

あるいは内包表記で書いても簡潔です。

>>> lst = [[1, 2, 3], [4, 5]]
>>> [e for row in lst for e in row]
[1, 2, 3, 4, 5]

hasattrはなぜ使われていたか？

要素が__iter__属性を持つかどうか、つまりイテラブルかどうか判定するために用いられています。

>>> lst = [1, 2, 3]
>>> hasattr(lst, '__iter__')
True
>>>
>>> tpl = (1, 2, 3)
>>> hasattr(tpl, '__iter__')
True
>>>
>>> num = 42
>>> hasattr(num, '__iter__')
False

Comments

[yu__]

非常に分かりやすかったです。ありがとうございました。
ベストアンサーにさせていただきました。

Answer 2

• flatten()の中でflatten()を呼び出しており、再帰関数になっているのは分かりますか？
• for item in data:でdataから要素を一つずつ取り出して繰り返し処理をしているのは分かりますか？

以上が分かっている前提で、
hasattr(item, '__iter__')はdataから取り出した要素itemが配列かどうかを確かめています。
配列だった場合は再起的に同関数を呼び出して繰り返しています。

Answer 3

if hasattr(item, 'iter'):

配列の要素が配列かどうか調べる

yield element

配列の要素を返す

ですので、配列の要素を順番に取り出して、もし要素が配列だった場合は再帰的に処理することで、その配列の要素も同様に取り出している。

という感じです。