データフレーム内のインデックスと列の名前が同じ場合、ラベルを選択する方法は？

Question

私はパンダのデータフレームを持っており、特定の行の非ゼロ要素の列のラベルを調べる必要があります。

ですから、例えば：コラムaについては

columns = ['a', 'b', 'c'] 
df = pd.DataFrame([[1, 0, 1], [0, 1, 0], [3, 5, 2]], index=columns, columns=columns) 

、私は['a', 'c']が返されることがしたいです。

Answer 1

これはお好みのものですか？

列をループし

df.index[(df != 0)['a']] 
# Index([u'a', u'c'], dtype='object') 

df.index[(df != 0)['b']] 
# Index([u'b', u'c'], dtype='object') 

df.index[(df != 0)['c']] 
# Index([u'a', u'c'], dtype='object') 

と辞書を構築、私たちが行うことができます。

map(lambda col: {col: df.index[(df != 0)[col]].values}, df.columns) 
# [{'a': array(['a', 'c'], dtype=object)}, 
# {'b': array(['b', 'c'], dtype=object)}, 
# {'c': array(['a', 'c'], dtype=object)}]