if-elif文で返り値が同じものを一つにまとめてかく簡潔な方法は？

以下、returnの返り値でA,Bが同じ値です。この各条件分岐内の返り値のA,Bを一つにまとめたいです。
一つの案として、返り値に含まないで、if文外で返すというふわっとしたイメージはありますが、どう書けばわからずにいます。

なにか、こうかけるなどのご提案頂けませんでしょうか？

python
1def XXX():
2    if LLL != PPP:
3       return 2, A, B
4
5    elif not T :
6        return 1,  A, B
7
8    elif D <= F:
9        return 1,  A, B
10
11    else:
12        return 0,  A, B
13

行動規範の内容に同意します

回答2件

ベストアンサー

python
1def XXX():
2    ret = [A, B]
3    if LLL != PPP:
4        s = 2
5    elif not T or D <= F:
6        s = 1
7    else:
8        s = 0
9    ret.insert(0, s)
10    return tuple(s)  # 返り値の型にこだわらないでsequence unpackして使うならtupleへの変換は不要

カンマ区切りで複数returnする際は、実態としてはtupleをreturnしているだけなのでこう書けます。

投稿2018/08/13 15:30

hayataka2049

総合スコア30933

sequelanonymous

2018/08/14 01:34

YYY = XXX(A, B, C) という感じで上記の関数の利用をかんがえていますが、tupleでもリストも引き渡せないような。。。どうすればよいでしょう。

hayataka2049

2018/08/14 05:12

そのことは簡略化とは関係ありませんが、もともとどうするつもりだったのですか

行動規範の内容に同意します

一瞬、三項演算子が浮かびましたが、見にくくなりますね。
関数を分けるのが見やすいと思います。

Python
1def XXX_sub():
2    if LLL != PPP:
3       return 2
4
5    elif not T :
6        return 1
7
8    elif D <= F:
9        return 1
10
11    else:
12        return 0
13
14def XXX():
15    return XXX_sub(), A, B