[英]Convert all columns in a Pandas DF to separate lists and assign it a name same as column name
[英]Convert pandas df cells to say column name
我有一个这样的 df:
0 1 2 3 4 5
abc 0 1 0 0 1
bcd 0 0 1 0 0
def 0 0 0 1 0
如果单元格中有 1,如何将数据框单元格转换为列名? 看起来像这样:
0 1 2 3 4 5
abc 0 2 0 0 5
bcd 0 0 3 0 0
def 0 0 0 4 0
让我们试试
df.loc[:,'1':] = df.loc[:,'1':] * df.columns[1:].astype(int)
df
Out[468]:
0 1 2 3 4 5
0 abc 0 2 0 0 5
1 bcd 0 0 3 0 0
2 def 0 0 0 4 0
我建议您简单地为每一列执行逻辑,其中给定列中的值为1
,将该值设置为列名
for col in df.columns:
df.loc[df[col] == 1, col] = col
我们可以在整个数据帧上使用np.where
:
values = np.where(df.eq(1), df.columns, df)
df = pd.DataFrame(values, columns=df.columns)
0 1 2 3 4 5
0 abc 0 2 0 0 5
1 bcd 0 0 3 0 0
2 def 0 0 0 4 0
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.