如何获得列名称为1的列。我还必须找出所有行是否只有一个1

Question

示例数据框

category    Nestle Britania P&G HL
    lead     0      1        0  0
    Mg       1      0        0  0
    Cl       0      0        1  0

首先，我们必须确保所有行都不超过1。

第二 所需的输出是一个数据框

category    company_name
 lead       britania
  Mg         Nestle
  Cl         P&G

我试图像下面那样

df['company_name'] = df.apply(lambda row: valuation_formula(row), axis=1)

def valuation_formula(row):
    return sent the column name which has result 1

请帮忙

Answer 1

使用DYLM_LIBRARY_PATH

dot