Создание столбца с условиями if-else-else в Pandas

Чтобы создать новый столбец на основе условия 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)) 

Этот подход более эффективен для больших наборов данных.

Вот пример использования предоставленного фрейма данных:

Входной кадр данных

Выходной кадр данных