代入先を条件で変えるif/elseをワンライナーで書きたい

Question

if文の結果によってparent.left, parent.rightのどちらの変数に代入するかが変わる4行の文を、ワンライナーでかくことはできるのでしょうか?
木構造を書いていて下記のような4行が度々出てくるので、短くなればいいなーと思っています。

if node.value < parent.value:
    parent.left = node.left
else:
    parent.right = node.left

# 下記のようなワンライナーは可能?
# (parent.left if node.value < parent.value else parent.right) = node.left

Comments

[meg_]

繰り返し出てくる処理なら関数化してはどうでしょうか？

[退会済みユーザー]

似たような構文が出てくるというだけで、条件や代入値に共通点があまりないのです。
そもそも、immutableなので...

[meg_]

if文の条件が変わるんですか。質問文からはそこまで読み取れずご希望に添えませんでした。

[quickquip]

> そもそも、immutableなので...
はどういう意味でしょう??
順当に考えたら parent.right = というような属性の書換を許さないということかと思ってしまうのですが。
(本筋とは関係ないのですが)

Answer 1

質問のように書くのが一番いいと思います。

パズルとして考えるなら

if node.value < parent.value:
    parent.left = node.left
else:
    parent.right = node.left

は

parent.left, parent.right, *_ = (parent.left, node.left, parent.right)[node.value < parent.value:]

と書き換え可能です。
（parent.left, parent.rightが属性としてもともと存在するという条件下で）

---

追記

クラスにsetterメソッドを用意して

(parent.set_left if node.value < parent.value else parent.set_right)(node.left)

とするのが1行で書けてかつ意味論的にもわかりやすい、ちょうどいい着地点かなと思いました。

Comments

[LouiS0616]

+10票くらい入れたいです。

[退会済みユーザー]

おおー、発想に脱帽です。

[quickquip]

思いついたことがあるので追記しました。(過分に+をもらっている気がします)

Answer 2

あまり行儀はよくありませんが、eval使うよりはマシかなと。

parent.__dict__['left' if node.value < parent.value else 'right'] = node.left

Comments

[LouiS0616]

文字列で評価するならsetattrの方が問題に即していて良いのかなと。

Answer 3

update(parent, node) としてメソッド定義して、
そのなかで、 if... else ... で parent の left, right を更新する処理を書くのが良いと思います。

Answer 4

代入先を変えるのは、１行は無理かもしれませんが２行なら、

if node.value < parent.value: parent.left = node.left
else: parent.right = node.left
```とか

Comments

[ForestSeo]

たまに簡単なif文の時、
例) if x == 5: b = 9
とか改行せずに書きます。

[退会済みユーザー]

blackによる強制フォーマットが入るので、結局改行されてしまうんです...

[ForestSeo]

ならそのままの方がいいんじゃないですか？

Answer 5

以下↓を参考に・・・
Python 三項演算子の書き方

Comments

[退会済みユーザー]

三項演算子は代入値には使えますが、代入先の変数の選択には使えません。
それに似たことができないか...というのが質問した理由です

[cateye]

あ！失礼しました。