pythonについて

Question

関数f(x)の戻り値a,bを一度変数に代入したりせずに、bのみを取り出したいです。どのように書けばできるのでしょうか…。

foo = [b**2 for x in bar for ??? in f(x)]  

in f(x)としてもf(x)からの戻り値はリストではないので駄目なんだな、というところまではわかります。
戻り値をリストにするか、うまく展開してあげる方法はありませんでしょうか？

Answer 1

pythonの関数が複数返り値を持つときはタプル形式で返すので、
「戻り値が2個であること」「片方しか使わないこと」が確実なのれあれば、
下のような処理はできたかと思います。

>>> def f(x):
...     return x+1, x*2
...
>>> ret = f(2)
>>> ret
(3, 4)
>>> type(ret)
<class 'tuple'>
>>> ret[1]
4

bがf()の戻り値のうち2個目であれば、こう書けはしました。

>>> bar = [1, 2, 3, 4, 5]
>>> foo = [f(x)[1] ** 2 for x in [1, 2, 3, 4, 5]]
>>> foo
[4, 16, 36, 64, 100]

Comments

[piyo56]

関数の戻り値ってタプルなんですね。勉強不足でした。どうもありがとうございます！