同じようなfor文とif文をの組み合わせをよりスマートにしたい。

Question

以下のようなfor文でif文の組み合わせがあった場合、よりスマートな書き方があると思っています。
しかし、中々別の書き方が思い浮かびません。こういう書き方もあるなどのアドバイス頂けませんでしょうか？

    for A_include in A_in_list:
       culculation = get_account(
            A_include, client, _id)
        if culculation is not None:
            not_list.append(culculation)

    for in in in_list:
        culculation = get_account(
            in, client, _id)
        if culculation is not None:
            not_list.append(culculation)

    for ex in ex_list:
        culculation = get_account(
            ex, client, _id)
        if culculation is not None:
            not_list.append(culculation)

Answer 1

inは予約語なので、変数名に使うとinvalid syntaxです・・・。

---

for A_include in A_in_list:
    culculation = get_account(
        A_include, client, _id)
    if culculation is not None:
        not_list.append(culculation)

for _in in in_list:
    culculation = get_account(
        _in, client, _id)
    if culculation is not None:
        not_list.append(culculation)

for ex in ex_list:
    culculation = get_account(
        ex, client, _id)
    if culculation is not None:
        not_list.append(culculation)

ループ対象とするリスト以外すべて同じ処理ということですね。

リストを結合してみますか。

for x in A_in_list + in_list + ex_list:
    culculation = get_account(x, client, _id)
    if culculation is not None:
        not_list.append(culculation)

これが気持ち悪ければ二重ループにしてあげることもできます。

for lst in [A_in_list, in_list, ex_list]:
    for x in lst:
        culculation = get_account(x, client, _id)
        if culculation is not None:
            not_list.append(culculation)

Answer 2

for のナカミを関数にまとめてしまえばいいんじゃないかと

ああ、ナカミといわず、for文自体を関数にまとめれますねー

Comments

[sequelanonymous]

すみません、どうまとめるべきかサンプルを念の為頂けませんでしょうか？

Answer 3

@y_waiwaiさん
for文自体を関数にまとめれますねー

まとめてみました。

import sys
print(sys.version)
hoge_list = ['hoge1','hoge2']
fuga_list = ['fuga1','fuga2']
piyo_list = ['piyo1','piyo2']

src_list = [hoge_list,fuga_list,piyo_list]
not_list = []
client = 0
_id = 0

def get_account(y, client, _id):
    print('test:',y, client, _id)

def func(src_list):
    for x in src_list:
        for y in x:
            culculation = get_account(y, client, _id)
            if culculation is not None:
                not_list.append(culculation)

func(src_list)

test: hoge1 0 0
test: hoge2 0 0
test: fuga1 0 0
test: fuga2 0 0
test: piyo1 0 0
test: piyo2 0 0