python 条件によって列計算の内容を変化させたい

Question

import pandas as pd
import numpy as np

ar = pd.DataFrame(np.random.randn(10, 3))
br = pd.DataFrame(np.random.randn(10, 3))

if ar.values > br.values:
    c = ar + br
else:
    d = ar + br + 10
    

arとbrの列の中の要素ごとに大小を比較して，arの要素の数値が大きい場合とbrの要素の数値が大きい場合とで計算式を変化させたいです．方法はありませんでしょうか．

上記コードでは，条件でcとdに分けていますが，分けないようしたいです．

以上，よろしくお願いします．

Answer 1

そもそも質問文のコードは動作しませんよね。

    if ar.values > br.values:
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

---

質問の意図を正しく汲めているかどうかどうかわかりませんが、こんな感じでしょうか。

import pandas as pd
import numpy as np

ar = pd.DataFrame(np.random.randn(5, 1))
br = pd.DataFrame(np.random.randn(5, 1))

result = ar + br
result[ar.values <= br.values] += 10
print("ar")
print(ar)

print("br")
print(br)

print("result")
print(result)
""" =>
ar
          0
0  0.011486
1  0.521872
2 -0.452988
3 -0.633430
4  0.505231
br
          0
0 -0.327572
1 -0.926398
2  0.084319
3  0.258488
4  2.290976
result
           0
0  -0.316087
1  -0.404526
2   9.631330
3   9.625057
4  12.796207
"""

Comments

[shougi]

そうですね，質問文のコードは動作しないです．説明するときにあったほうがわかりやすいかと思い，載せました．

なるほど，こういう書き方があるのですね．本当にありがとうございます！