Python リストに格納した関数を、別のリストにおいてインデックスが同じ位置にある変数がTrueの時だけ実行したい

Question

リストに格納した関数を、別のリストにおいてインデックスが同じ位置にある変数がTrueの時だけ実行したいです。
ただ、こうした考え方が、下記で例に挙げていることをしたい解として、正しいかもわかりかねております。
考え方を変える必要がある場合は、その点もご教示頂けますと大変助かります。

例：1という数値が含まれるリスト（numbers）が与えられた際に、数値を2倍する関数で得られた[1,2]というリストに対して、次に数値を3倍する関数を適用し、[1, 2, 3, 6]という結果を得たい。ただし、別のリスト（methods）の同じインデックス位置がtrueの時だけ実行したい。

■インプット
[1]

■得たい結果
・bai2, bai3がいずれもTrueの場合　→[1, 2, 3, 6]
・bai2=False, bai3=Trueの場合　→[1, 3]
・bai2=True, bai3=Falseの場合　→[1, 2]

import numpy as np

num = int(1)

# 得たいリストを計算する関数の定義
def scratch_num(num, bai2=False, bai3=True):
    
    # booleanのリスト
    methods = [bai2, bai3]
    
    # 関数用リスト
    function = np.array([
    lambda x: x * 2, 
    lambda x: x * 3
    ])

    # 結果リスト
    numbers = [num]

    # 結果リストに対して、関数リストを適用
    for f in function:

        tmp = []  # f の適用結果を記録する一時的な配列
        for x in numbers:
            y = f(x)
            tmp.append(y)

        numbers.extend(tmp)
    
    return numbers

# 関数の実行
numbers = scratch_num(num)

# 結果の出力
print(numbers)

Comments

[TakaiY]

質問は何ですか？

- アルゴリズムとして正しいかどうか？
ある入力に対して想定通りの結果が得られるのであれば、正しいでしょう。
得られないのであれば、どのような入力でどのような結果が出てしまうのか追記してください。

- 他のアルゴリズムが知りたいですか？
他のやりかたはたくさんありますが、ここはそういう漠然としたことを聞く場所ではないので、
もっと考えてから質問したほうがいいですよ。

[yukimz]

@TakaiYさま
追記いただきありがとうございます。質問は、現状のコードでは、関数と真偽値の紐づけが出来ていない（どうやるかがわからない）ため、それをどのように行うか？（コードのどの部分を直すか？）ということでした。何を聞きたいかを明確にして質問するように留意いたします。

Answer 1

こんな感じでどうでしょう？

# 関数
bai2 = lambda x: x * 2
bai3 = lambda x: x * 3
plus1 = lambda x: x + 1
plus2 = lambda x: x + 2
minus5 = lambda x: x - 5

# 関数のリスト
funcs = [bai2, bai3, plus1, plus2, minus5]

# 関数を実行するかしないかのフラグのリスト
flags = [True, False, True, False, True]  # => bai2, plus1, minus5 を実行する

# 関数を適用するリストの初期値
numbers = [1]

# funcs と flags を zip で組み合わせてループさせる
for func, flag in zip(funcs, flags):
    if flag:
        numbers += [func(x) for x in numbers]  # => [1, 2, 2, 3, -4, -3, -3, -2]


print(numbers)

• 動作確認用CodePen:https://repl.it/@jun68ykt/Q252734

Comments

[yukimz]

リストを結合して、Trueの場合だけ実行させる、というのを、zipで実現するんですね！
ありがとうございます！大変助かりました。

[jun68ykt]

どういたしまして
> 大変助かりました。
とのことで、よかったです????

Answer 2

その後、自己解決した方法としては、質問のコードの下記部分の修正でいけました！

    for f in function[methods]: #もともとは"for f in functionとしていたが、[methods]を追加することで、Trueの場合のみ実行可能となった。

        tmp = []  # f の適用結果を記録する一時的な配列
        for x in images:
            y = f(x)
            tmp.append(y)

        images.extend(tmp)