在 Pandas 中使用 If-Else-Else 條件建立列

根據 if-elif-else條件建立新列，主要有兩種方法：

非向量化方法

此方法涉及定義一個對行進行操作的函數：

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)) 

此方法對於大型資料集更有效。

這是使用提供的資料幀的範例：

A

輸出DataFrame

-1