python - 如何 "denormalize"数据框/从许多二进制列中创建一列

Question

我有一个包含许多二进制列的数据框，这些列指示是否提到了特定的产品名称。我想创建一个单独的列来列出所有这些特定的产品名称，其中该行带有 1。

为简单起见，假设这是我的数据框:

df = pd.DataFrame({'Name': [1,0,0], 'Another Name': [0,1,1], 'Different Name':[0,0,1]})

我想创建这个专栏:

0 ['Name']
1 ['Another Name']
2 ['Another Name','Different Name']

我的思考过程是遍历每一行，如果任何名称有 1，则将其添加到该列的列表中

namelist = list()
if df['Name']==1:
    namelist.append("Name")
else if df['Another Name']==1:
    namelist.append("Another Name")
else if df['Different Name']==1:
    namelist.append("Different Name") 

但这不会使列表特定于行。有关如何执行此操作的建议？

我的解决方案:我使用了 G. Anderson 解决方案中的逻辑，但我需要指定感兴趣的列，而不是数据框中的所有列。我确信有比我最终做的更好的方法来做到这一点，但这就是我所做的:

df['Name']=df['Name'].replace({1:'Name',0:''})
df['Another Name']=df['Another Name'].replace({1:'Another Name',0:''})
df['Different Name']=df['Different Name'].replace({1:'Different Name',0:''})

df['Product Name']=df['Name'] + df['Another Name'] + df['Different Name']

Answer 1

这是我的镜头:

df = pd.DataFrame({'Name': (1,0,0), 'Another Name': [0,1,1], 'Different Name':[0,0,1]})

    Name    Another Name    Different Name
0   1       0               0
1   0       1               0
2   0       1               1

用列名或''

替换值

for col in df.columns:
    df[col]=df[col].replace({1:col,0:''})

    Name    Another Name    Different Name
0   Name        
1           Another Name    
2           Another Name    Different Name

添加一列，该列是其他列值的列表

df['new_col']=df.iloc[:,:].apply(lambda x: [i for i in list(x) if i], axis=1)

    Name    Another Name    Different Name  new_col
0   Name                                    [Name]
1           Another Name                    [Another Name]
2           Another Name    Different Name  [Another Name, Different Name]

删除其他列

df=df['new_col']

0                            [Name]
1                    [Another Name]
2    [Another Name, Different Name]
Name: new_col, dtype: object