Найти столбцы с истинным значением

Question

У меня есть DataFrame со столбцами, что в строке есть только True (как df ниже).Как я могу сформировать столбец с индексом столбца, который имеет True для этой строки?В реальной жизни матрица большая, поэтому я избегаю использования for loop и apply.

df = pd.DataFrame([[True,False,False],
                   [True,False,False],
                   [False,True,False],
                   [False,False,True],
                   [True,False,False],
                   ])

Ответ будет выглядеть примерно так:

df['TrueColIndex'] = [0,0,1,2,1]

Спасибо!

Answer 1

Это idxmax

df.idxmax(1)
Out[444]: 
0    0
1    0
2    1
3    2
4    0
dtype: int64