Как добавить значение в новый столбец исходя из данных другого столбца?

Question

[idr1995]

Из таблицы НБА вывел два столбца:
Первый столбец - игрок,
Второй столбец - его команда.

Создаю третий столбец (Conference). Нужно добавить значения в этот столбец. Всего значений два - это Eastern и Western.
Исходя из столбца Team не могу добавить эти два значения под подходящие команды.

Весь код получился следующий:

nba = pd.read_csv('nba.csv', usecols=['Name', 'Team'])
teams = nba['Team'].unique()
west = ['Boston Celtics', 'Brooklyn Nets', 'New York Knicks',
       'Philadelphia 76ers', 'Toronto Raptors', 'Chicago Bulls', 'Cleveland Cavaliers',
       'Detroit Pistons', 'Indiana Pacers', 'Milwaukee Bucks', 'Charlotte Hornets',
       'Miami Heat', 'Orlando Magic',
       'Washington Wizards', 'Minnesota Timberwolves']

for i in nba['Team']:
    if i in west:
        nba['conference'] = 'western'
    else:
        nba['conference'] = 'eastern'

но в итоге у меня получается, что весь столбец состоит из значения Eastern

Comments

[Алан Гибизов]

См.п.3.8

[Дмитрий]

Скорее всего, вам стоит глазами проверить что находится в teams и что находится в nba['Team']

Answer 1

[dmshar]

Ну, полностью повтроить ваш пример, разумеется не получится, но решение - вот оно:

list_a = [["Федя",'Boston Celtics'],
          ["Петя",'Toronto Raptors'],
          ["Коля",'Toronto Raptors'],
          ["Вася", 'Молодые таланты'],
          ["Онуфрий",'Старые инвалиды']]

columns = ['Name', 'Team']
nba = pd.DataFrame(list_a, columns=columns)

west = ['Boston Celtics', 'Brooklyn Nets', 'New York Knicks',
       'Philadelphia 76ers', 'Toronto Raptors', 'Chicago Bulls', 'Cleveland Cavaliers',
       'Detroit Pistons', 'Indiana Pacers', 'Milwaukee Bucks', 'Charlotte Hornets',
       'Miami Heat', 'Orlando Magic',
       'Washington Wizards', 'Minnesota Timberwolves']

nba['conference']=np.where(nba['Team'].isin(west), 'western','eastern' )
print(nba)

Результат:

Name             Team conference
0     Федя   Boston Celtics    western
1     Петя  Toronto Raptors    western
2     Коля  Toronto Raptors    western
3     Вася  Молодые таланты    eastern
4  Онуфрий  Старые инвалиды    eastern

Comments

[idr1995]

О, спасибо большое вам за развернутый ответ!)))
Посмотрел код, все не так сложно оказывается, спасибо)

[dmshar]

idr1995, Стесняюсь спросить. Посмотрел ваши вопросы и ответы на них. Ни один из них вы не приняли как ответ, который вам помог. А люди старались и намой взгляд большинство ответов - полезные. Это ваша принципиальная позиция?

[idr1995]

dmshar, честно признаюсь, я не знал, что здесь надо отмечать ответ как решение. Исправился