Повторяющиеся строки в pandas DF - хотите вернуть список уникальных строк

Question

У меня есть группа строк данных в pandas DF, которые содержат несогласованно компенсирующие строковые символы. Для каждого ИД игры (другой столбец) два строковых символа уникальны для этого Идентификатора игры, но не выключаются в предикативном шаблоне. Независимо от того, я пытаюсь написать вспомогательную функцию, которая принимает каждый уникальный идентификатор игры и получает имена двух команд, связанных с ним.

Например ...

index game_id 0 400827888 1 400827888 2 400827888 3 400827888 4 400827888 ... 555622 400829117 555623 400829117 555624 400829117 555625 400829117

index team 0 ATL 1 DET 2 ATL 3 DET 4 ATL ... 555622 POR 555623 DEN 555624 POR 555625 POR

Вот мой горестный попытка, которая не работает.

def get_teams(df): 
    for i in df['gameid']: 
     both_teams = [df['team'].astype(str)] 
     return(both_teams) 

Я хотел бы, чтобы вернуться [ 'ATL', 'DET] для игры ID 400827888 и [' POR», 'DEN'] для игры ID 400829117. Вместо этого он просто возвращается имя команды связанных с каждым индексом.

Answer 1

Вы можете использовать SeriesGroupBy.unique:

print (df.groupby('game_id')['team'].unique()) 
game_id 
400827888 [ATL, DET] 
400829117 [POR, DEN] 
Name: team, dtype: object 

Для цикла использования iterrows:

for i, g in df.groupby('game_id')['team'].unique().reset_index().iterrows(): 
    print (g.game_id) 
    print (g.team) 

EDIT:

Если нужно найти все game_id по некоторой строке (например DET) использовать boolean indexing:

s = df.groupby('game_id')['team'].unique() 

print (s[s.apply(lambda x: 'DET' in x)].index.tolist()) 
[400827888]