python - Pandas : Data Frame Pruning

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 列采用逻辑 OR。ID 和 NAME 始终具有相同的对值，无论它们出现多少次，但 A 列和 B 列可以更改(00,10,11,01)。

所以最后我想要 ID、NAME、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()