Python pandas データフレームのカラム追加(条件判定の計算あり)について

Question

環境：python:3.5.1, pandas:0.17.1

import pandas as pd
df=pd.DataFrame([[15,25,35,40],[25,10,40,55],[-10,15,-5,-15]]).T
df.columns=["A","B","diff"]

"diff"は"A"-"B"の計算結果です。
各インデックスで"diff"が正ならば1、"diff"が負ならば-1の値を取る新しいカラム"class"をdfに追加したいですが、方法がわかりません。ご教授のほどよろしくお願いします。

df
A B diff
15 25 -10
25 10 15
35 40 -5
40 55 -15

やりたいこと
df
A B diff class
15 25 -10 -1
25 10 15 1
35 40 -5 -1
40 55 -15 -1

Answer 1

以下で、やりたいことにマッチしますでしょうか？

df["class"] = df["diff"].apply( lambda x: -1 if x < 0 else 1 )

Comments

[zone99exit]

回答ありがとうございます。ご指摘の方法で出来ました。
条件判定のやり方がわかったので一歩先に進めました。

[退会済みユーザー]

自分のものよりシンプルでスマートですね。勉強になりました。

Answer 2

いくつかやり方はあります。

一つのやり方としてはdiffの絶対値でdiffの値を割りましょう。
ただし0で割る場合NaNになってしまうので、
その部分をあとから0に変換します。

import numpy as np
df["class"] = np.true_divide(df["diff"], np.abs(df["diff"]))
df["class"] = df["class"].fillna(0)

Comments

[zone99exit]

回答ありがとうございます。ご指摘のやり方で出来ました。