根據數據框列之間的條件生成列名字典

Question

我有以下數據框：

                        a_11        b_14    c_13     d_12
AC                      True        False   False   False 
BA                      True        False   False   True
AA                      False       False   False   False 

我想要一個字典，鍵作為索引，值作為具有真實值的列名列表，即

{
AC : [a_11],
BA : [a_11,d_12],
AA : []
}

我應該如何處理這個問題

編輯：列名是字符串，而不是字符。

Answer 1

如果性能對轉置 DataFrame 很重要並將列名稱轉換為列表，請使用字典理解：

d = {k: v.index[v].tolist() for k, v in df.T.items()}
print (d)
{'AC': ['a_11'], 'BA': ['a_11', 'd_12'], 'AA': []}

另一個想法zip並通過DataFrame.to_numpy將值轉換為二維 numpy 數組：

d = {k: df.columns[v].tolist() for k, v in zip(df.index, df.to_numpy())}
print (d)
{'AC': ['a_11'], 'BA': ['a_11', 'd_12'], 'AA': []}

Answer 2

您可以在此處使用df.mul將df與df.columns相乘，然后使用df.agg過濾掉空字符串''

out = df.mul(df.columns).agg(lambda x:[*filter(None, x)], axis=1)

AC          [a_11]
BA    [a_11, d_12]
AA              []
dtype: object

您可以在此處使用列表理解。

vals = [df.columns[m].tolist() for m in df.values]
# vals -> [['a_11'], ['a_11', 'd_12'], []]
pd.Series(vals, index=df.index)

AC          [a_11]
BA    [a_11, d_12]
AA              []
dtype: object