pythonで複数の返り値をそのまま複数のリストにappendする方法

Question

関数から返り値が複数あり、以下の様にリストに追加することをやりたいのですが、わざわざ変数を作らずにappendする方法はないかと思い質問しました。

def hoge()
    return [1,2,3], [4,5,6], [7,8,9]

a = []
b = []
c = []

list1, list2, list3 = hoge()
a.append(list1)
b.append(list2)
c.append(list3)

list1～3を作らず同じ動作をする賢い書き方ないですかね

Answer 1

ちょっと無理やり感がありますが、以下でどうでしょうか。

def hoge():
    return [1,2,3], [4,5,6], [7,8,9]

a,b,c = [],[],[]
for i,r in zip([a,b,c],hoge()):
    i.append(r)

print(a)# [[1, 2, 3]]
print(b)# [[4, 5, 6]]
print(c)# [[7, 8, 9]]

Comments

[Amanokawa]

回答ありがとうございます。zipを使うことにします。
a,b,cの作り方はまとめた方がすっきりしますね。

Answer 2

賢いかどうかは微妙ですけど、一つの案として。

for lst, e in zip([a, b, c], hoge()):
    lst.append(e)

なお、ご提示の例に限れば次のようにも書けます。

a, b, c = map(lambda l: [l], hoge())

Comments

[Amanokawa]

なるほど、zipを使えばいいのですね。
ありがとうございます。

Answer 3

enumerate や zip をつかってみました。
a.py

def hoge():
    return [1,2,3], [4,5,6], [7,8,9]

a = ['a']
b = ['b']
c = ['c']

[[a, b, c][idx].append(el) for idx, el in enumerate(hoge())]
print(a)
print(b)
print(c)

[src.append(el) for src, el in zip([a, b, c], hoge())]
print(a)
print(b)
print(c)

実行例

参考情報

• Python, enumerateの使い方: リストの要素とインデックスを取得

https://note.nkmk.me/python-enumerate-start/

• Python, zip関数の使い方

https://note.nkmk.me/python-zip-usage-for/