新しい列を追加する際の関数の指定について

Question

以下のようなデータフレームに「Life Ladder」の値が4.5以上ならhappy,それ以下ならuhappyという文字列を入れる「happy_or_unhappy」という列を追加したいのですが、
「ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().」
というエラーが出ます。
関数の指定の仕方が間違えているのだと思いますが、間違えている箇所を教えていただけると幸いです。

↓コードです

def happy_or_unhappy(self):
    if hp2["Life Ladder"] >= 4.5:
        return "happy"
    else:
        return "unhappy"
    
hp2['happy_or_unhappy'] = hp2["Life Ladder"].apply(happy_or_unhappy)

Answer 1

hp2["Life Ladder"]がSeriesなので、if文の条件式の中がboolの配列になっています。

---

追記
例えば関数を、

def happy_or_unhappy(x):
    if x >= 4.5:
        return "happy"
    else:
        return "unhappy"

一行で書くなら、

hp2['happy_or_unhappy'] = hp2["Life Ladder"].apply(lambda x: 'happy' if x>=4.5 else 'unhappy')

Comments

[renren643]

回答ありがとうございます。できました。
bool値になっているとどのように関数の指定の仕方が変わってくるのですか？
申し訳ありませんが、私の知識が不足しており、もう少し詳しく教えていただけると幸いです。

[mkgrei]

hp2["Life Ladder"]は[1.4, 5.1, 6.3, ...]みたいになっているので、
hp2["Life Ladder"]>=4.5は[False, True, True, ...]のbool値を要素に持つ配列になります。

そこが問題になっていて、上記のエラーはたくさんのbool値を与えられたので、どれをif文の判定にすべきかわからないから困った、と言っています。
.applyはそれぞれの要素を取り出しているので、その取り出された要素に対して比較を行えば良いことになります。

[renren643]

詳しい回答、解説ありがとうございました！スッキリしました！

Answer 2

np.where() を使う方法もあります

hp2['happy_or_unhappy'] = np.where(hp2["Life Ladder"] >= 4.5, 'happy', 'unhappy')