Pandas で If-Else-Else 条件を使用した列を作成する

if-elif-else 条件に基づいて新しい列を作成するには2 つの主なアプローチがあります:

非ベクトル化アプローチ

このアプローチには、行を操作する関数の定義が含まれます:

def f(row):
    if row['A'] == row['B']:
        val = 0
    elif row['A'] > row['B']:
        val = 1
    else:
        val = -1
    return val

次に、行に沿ってデータフレームに適用します:

df['C'] = df.apply(f, axis=1)

ベクトル化されたアプローチ

ベクトル化されたアプローチでは、np.where を利用して新しい列を直接作成します。

df['C'] = np.where(
    df['A'] == df['B'], 0, np.where(
    df['A'] >  df['B'], 1, -1)) 

このアプローチは、大規模なデータセットの場合により効率的です。

提供されたデータフレームを使用した例を次に示します。

入力データフレーム

出力データフレーム