如何获取列名称1?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (1)
  • 关注 (0)
  • 查看 (34)

示例数据帧

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

你能帮忙吗?

提问于
用户回答回答于

运用 dot

df['New']=df.iloc[:,1:].dot(df.columns[1:])
df
Out[106]: 
  category  Nestle  Britania  P&G  HL       New
0     lead       0         1    0   0  Britania
1       Mg       1         0    0   0    Nestle
2       Cl       0         0    1   0       P&G

所属标签

可能回答问题的人

  • uncle_light

    5 粉丝518 提问7 回答
  • 学生

    3 粉丝476 提问7 回答
  • o o

    4 粉丝494 提问5 回答
  • 富有想象力的人

    3 粉丝0 提问5 回答

扫码关注云+社区

领取腾讯云代金券