python3のジェネレーターについて

Question

以下の関数をpython3でジェネレーターに変更したいです。

def mufunc(start, end, multiple, non_multiple):
print(*filter(lambda i: i % multiple == 0 != i % non_multiple, range(start, end + 1)))

Answer 1

def mufunc(start, end, multiple, non_multiple):
    yield from filter(lambda i: i % multiple == 0 != i % non_multiple, range(start, end + 1))

Answer 2

補足の補足という意味の分からないことをしますが、KSwordOfHasteさんがいろいろ調べたことに対して。
yield fromを使うと、yield fromを複数並べることが出来るようになるりますね。

def mufunc(start, end, multiple, non_multiple):
    for i in range(2):
        yield from filter(lambda i: i % multiple == 0 != i % non_multiple, 
    # 上のfor文は次と2行と同等
    # yield from filter(lambda i: i % multiple == 0 != i % non_multiple, 
    # yield from filter(lambda i: i % multiple == 0 != i % non_multiple, 


range(start, end + 1))

yieldやyield fromはもっと複雑な使い方も出来るのですが、あんまりおもしろい例を知らないので、ひとまずreturn filter(xxx)との違いの補足までに。

Comments

[KSwordOfHaste]

generatorの柔軟な点は規則性がない(不連続な)列や個々の要素を自由に一つの列として合成できる点にあると思います。

def concat(*iterators):
　for iterator in iterators:
　　yield from iterator

こんなのもすっきり書けるというのが便利ですよね。ちなみに他の言語でもyield from的な構文あると思います。Pythonにもあるだろうと気づいてしかるべきでした。アハハ(汗

Answer 3

質問内容の本質には関係なさそうな話ですが・・・

YouheiSakuraiさんの回答をみて「yield from」とかけることを知り、ジェネレーターの書き方を調べているうちに結構色々書き方があることがわかったので、速度を簡単に比べてみました。

import time, math

# 様々な列を生成できるのがgeneratorの基本でしょうから
# yield fromを知らなかったのは損してました

def f1(start, end, multiple, non_multiple):
    yield from filter(lambda i: i % multiple == 0 != i % non_multiple,
                      range(start, end + 1))

# 素朴な例?

def f2(start, end, multiple, non_multiple):
    for i in range(start, end + 1):
        if i % multiple == 0 != i % non_multiple:
            yield i

# 内包表記と同じようにジェネレーターを書けるということを初めて知りました!

def f3(start, end, multiple, non_multiple):
    return (i for i in range(start, end + 1)
              if i % multiple == 0 != i % non_multiple)

# これはgeneratorではないかも知れないが使い勝手は同じだろうと思ったので比較対象に入れた

def f4(start, end, multiple, non_multiple):
    return filter(lambda i: i % multiple == 0 != i % non_multiple,
                  range(start, end + 1))

start = 0
end = 1000000
multiple = 2
non_multiple = 3

for f in [f1, f2, f3, f4, f1, f2, f3, f4]:
    s1 = 0
    s2 = 0
    n = 20
    for i in range(n):
        t = time.time()
        sum(f(start, end, multiple, non_multiple))
        t = time.time() - t
        s1 += t
        s2 += t * t
    μ = s1 / n
    σ = math.sqrt(s2 / n - μ * μ)
    print("{}: {:.6f} {:.6f}".format(f.__name__, μ, σ))

各々sumを20回実行し平均、標準偏差を印字しています。実行時間がかなり変動するので2セットやってます。

f1: 0.280532 0.034435
f2: 0.147682 0.007799
f3: 0.147200 0.009094
f4: 0.195394 0.007899
f1: 0.214874 0.006771
f2: 0.146917 0.007629
f3: 0.149237 0.009201
f4: 0.198542 0.007140
(Windows 10 64bit, Python 3.6.0)

測ったくせしてなんですが、非常に些細な差なので「わかりやすい」と感じるものを選ぶのがやはりいいのかなと思いました。

---

ちなみにfilterで機能が達成できているのですから、それを**わざわざgeneratorにする必要性はない気がしました。**どうなのでしょう？

Comments

[YouheiSakurai]

f4ではなくf1にした理由は、f1の方がなんとなく見た目がジェネレーターっぽいからだと思います。正直、全然深くは考えていませんでした。

[KSwordOfHaste]

generatorでyieldとyield fromを使い分けてさまざまな列を柔軟に生成できるんだな！と言う点に気づくことができてYouheiSakuraiさんの回答で「これはよいことを知った」と自分は思いました。

[退会済みユーザー]

filter(function, iterable) は(item for item in iterable if function(item))と同じということなので、妥当なご指摘みたいですね。https://docs.python.org/3/library/functions.html#filter

[KSwordOfHaste]

dkato0077さん
補足ありがとうございました。なるほど書いてありますね！
今まで内包表記と同じノリでgeneratorが書けるといったことも知らなかったので本件でよい勉強させていただきました。