Панды: Обрезка Фрейма Данных

Question

У меня есть фрейм данных, как указано ниже:

data = [['1','tom',1,0],['1','tom',0,1],['2','lynda',0,1],['2','lynda',0,1]]
df = pd.DataFrame(data, columns = ['ID','NAME', 'A','B'])

df.head()

Я хочу преобразовать фрейм данных, чтобы он выглядел так, как показано ниже: где влогическое ИЛИ берется для столбцов A и B. ID и NAME всегда будут иметь одинаковые значения пары независимо от того, сколько раз они появляются, но столбцы A и B могут изменяться (00,10,11,01).

Итак, в конце я хочу ID, имя, A, B.

Answer 1

Вы всегда можете сложить и сравнить с 0.

data = [['1','tom',1,0],['1','tom',0,1],['2','lynda',0,1],['2','lynda',0,1]]
df = pd.DataFrame(data, columns = ['ID','NAME', 'A','B'])

g_df = (df.groupby(['ID', 'NAME']).sum() >0).astype(float)

g_df.reset_index()