Создание столбца с помощью операторов if else

У меня есть фрейм данных с 4 столбцами. Столбцы: Домашняя_команда Команда_гости Домашняя_команда_гол Away_team_goal

Я хочу создать столбец победителей, используя цели. Я пишу код ниже, но когда я его запускаю, в столбце победителей появляются только домашние команды.


A = ['A', '1', 'D', '2']
B = ['B', '2', 'E', '2']
C = ['C', '3', 'F', '2']

df = pd.DataFrame(np.array([A,B,C]), columns = ['Home_team', 'Home_team_goal', 'Away_team', 'Away_team_goal'])

df['Home_team_goal'] = pd.to_numeric(df['Home_team_goal'])
df['Away_team_goal'] = pd.to_numeric(df['Away_team_goal'])



df['Average'] = df['Home_team_goal'] - df['Away_team_goal']

df['Winner'] = ''

for i, e in enumerate(df['Average']):
    if e > 0:
        df['Winner'] = df['Home_team']
    elif e < 0:
        df['Winner'] = df['Away_team']
    else:
        df['Winner'] = 'Tie'

Пожалуйста, прочитайте минимальный воспроизводимый пример

MatBailie 28.08.2024 23:39

Используйте np.where().

Barmar 28.08.2024 23:50
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
2
51
3
Перейти к ответу Данный вопрос помечен как решенный

Ответы 3

Это можно сделать с помощью Numpy.select:

import numpy as np
cond1 = df['Home_team_goal'] > df['Away_team_goal']
cond2 = df['Home_team_goal'] < df['Away_team_goal']

df['Winner'] = np.select([cond1, cond2], ['Home_team', 'Away_team'], default = 'Tie')

дополнительно (напечатал здесь как слишком неудобно для комментария....

df['Average'] = df['Home_team_goal'] - df['Away_team_goal']

бросит TypeError: unsupported operand type(s) for -: 'str' and 'str'

вам нужно преобразовать их в целые числа

df['Home_team_goal'] = df['Home_team_goal'].astype(int)
df['Away_team_goal'] = df['Away_team_goal'].astype(int)
Ответ принят как подходящий

Здесь другой подход:

def Winner(x):
if x > 0:
    return 1
elif x < 0:
    return 2
else:
    return 0

df['Winner'] = df['Average'].apply(Winner)

Другие вопросы по теме