在pandas df中返回列名称的最有效方法

用途

Dataframe.dot

df.astype(bool).dot(df.columns).str.cat(sep=',')

要么，

','.join(df.astype(bool).dot(df.columns))'A,C,B,A'

或者，作为列表：

df.astype(bool).dot(df.columns).tolist()['A', 'C', 'B', 'A']

…或系列：

df.astype(bool).dot(df.columns)0    A1    C2    B3    Adtype: object