Панды - выглядящие значения в двух столбцах

Question

У меня есть фрейм данных с четырьмя столбцами.

• Team1
• Team2
• Победитель
• Host_Country

Теперь я хочу добавить новый столбец «Home_Away», чтобы заполнить «Home» или «Away» в зависимости от того, совпадает ли «Winner» с «Host_Country».Конечный результат должен быть:

• Team1
• Team2
• Победитель
• Host_Country
• Home_Away

Пожалуйста, кто-нибудь может мне помочь, как мне это сделать.Спасибо

Answer 1

ищите решение вашего вопроса.

import pandas as pd
import numpy as np

df = pd.DataFrame({'Team1': ['Black Eagles',
                             'Banana Slugs',
                             'Preachers',
                             'Fighting Cardinals',
                             'The Predators',
                             'Razorbacks',
                             'Rebels',
                             'Fighting Crusaders',
                             'Avengers',
                             'Aztecs'],
                   'Team2': ['Preachers',
                             'Fighting Cardinals',
                             'The Predators',
                             'Razorbacks',
                             'Rebels',
                             'Fighting Crusaders',
                             'Black Eagles',
                             'Banana Slugs',
                             'Avengers',
                             'Aztecs'],
                   'Winner': ['Preachers',
                              'Fighting Cardinals',
                              'The Predators',
                              'Razorbacks',
                              'Rebels',
                              'Fighting Crusaders',
                              'Black Eagles',
                              'Banana Slugs',
                              'Avengers',
                              'Aztecs'],
                   'Host_Country': ['Black Eagles',
                                    'Banana Slugs',
                                    'Preachers',
                                    'Fighting Cardinals',
                                    'The Predators',
                                    'Razorbacks',
                                    'Rebels',
                                    'Fighting Crusaders',
                                    'Avengers',
                                    'Aztecs']})

df = df[['Team1', 'Team2', 'Winner', 'Host_Country']]

df['Home_Away'] = np.where((df['Winner'] == df['Host_Country']), 'Home', 'Away')

print(df)

и результат будет выглядеть следующим образом

               Team1               Team2              Winner                Host_Country        Home_Away  
0        Black Eagles           Preachers           Preachers               Black Eagles        Away  
1        Banana Slugs  Fighting Cardinals  Fighting Cardinals               Banana Slugs        Away  
2           Preachers       The Predators       The Predators                  Preachers        Away  
3  Fighting Cardinals          Razorbacks          Razorbacks         Fighting Cardinals        Away  
4       The Predators              Rebels              Rebels              The Predators        Away  
5          Razorbacks  Fighting Crusaders  Fighting Crusaders                 Razorbacks        Away  
6              Rebels        Black Eagles        Black Eagles                     Rebels        Away  
7  Fighting Crusaders        Banana Slugs        Banana Slugs         Fighting Crusaders        Away  
8            Avengers            Avengers            Avengers                   Avengers        Home  
9              Aztecs              Aztecs              Aztecs                     Aztecs        Home