python - 如果值= 1，则pandas数据框打印列

Question

将此数据放入 panda 数据框中:

animal      tall    swim        jump  
frog          0        1            1
toad          1        0            0
tadpole       0        0            1

想要这样的输出:

frog swim
frog jump
toad tall
tadpole jump

有任何内置函数吗？

Answer 1

使用set_index与 stack ，过滤 1 并通过构造函数创建 DataFrame:

df1 = df.set_index('animal').stack()
s = df1[df1==1]

df2 = pd.DataFrame({'a':s.index.get_level_values(0),
                    'b':s.index.get_level_values(1)})
print (df2)
         a     b
0     frog  swim
1     frog  jump
2     toad  tall
3  tadpole  jump

另一个解决方案是使用MultiIndex.to_frame :

df2 = s.index.to_frame(index=False)
df2.columns = ['a','b']
print (df2)
         a     b
0     frog  swim
1     frog  jump
2     toad  tall
3  tadpole  jump

melt 的不同解决方案并按 query 进行过滤:

df2 = (df.melt('animal', var_name='val')
         .query('value == 1')
         .sort_values('animal')
         .reset_index(drop=True)
         .drop('value', axis=1))