データフレームへの関数適用

Question

python データフレームにて、行毎に関数を適用したいです。

行数は任意で、列は4列あります。
1列目と2列目がともに0かつ、3列目または4列目に0以外が入っている場合に1を返し、それ以外は0を返す関数を作ろうと
しております。
返した値をさらにデータフレームに格納したいです。

関数を作成して、その後、データフレームの各行毎に適用する方法を教えていただけないでしょうか。

#対象データフレームの例
df = DataFrame([[0,0,0,0],[10,12,10,12],[0,0,11,15],[0,0,0,0],[0,0,0,12]],index=["あ","い","う","え","お"],columns=["a","b","A","B"])

#適用させたい関数
if (df["a"][0]==0)&(df["b"][0]==0)&((df["A"]!=0)|(df["B"]!=0)):
    print(1)
else:
    print(0)

最終的に、（0,0,1,0,1）という5行1列のデータフレームを出力したいです。

なにとぞよろしくお願いいたします。

Answer 1

行毎に関数適用しなくても、単に

import pandas as pd

df = pd.DataFrame([[0,0,0,0],[10,12,10,12],[0,0,11,15],[0,0,0,0],[0,0,0,12]],index=["あ","い","う","え","お"],columns=["a","b","A","B"])

df['ret'] = ((df.a == 0) & (df.b == 0) & (df.A != 0) & (df.B != 0)).astype('int8')

でよいのでは？

Comments

[yuyu127]

ありがとうございました。
適用することができました。

Answer 2

行、列に対しての操作には.apply()を使っています。

df.apply(lambda x: (x["a"] == 0) & (x["b"] == 0) & ((x["A"]!=0) | (x["B"]!=0)), axis=1).astype(int)

# or 

def func(x):
    return (x["a"] == 0) & (x["b"] == 0) & ((x["A"]!=0) | (x["B"]!=0))
df.apply(func, axis=1).astype(int)
```

Comments

[yuyu127]

ありがとうございます。
.apply()でもうまくいきました。

Answer 3

http://sinhrks.hatenablog.com/entry/2015/06/18/221747

Comments

[yuyu127]

情報ありがとうございます。
助かりました。