python - 如何将多列转换为单列？

Question

我有一个onehot-encoded列，df，零为“nan”。我正在尝试将 onehot 编码的列转换为单列。

假设下面的数据框，df

    p1   |   p2  |   p3   |  p4   |  p5   |
---------------------------------------
0   cat     nan     nan     nan      nan
1   nan     dog     nan     nan      nan
2   nan     nan     horse   nan      nan
3   nan     nan     nan     donkey   nan
4   nan     nan     nan     nan      pig   

所需输出

    animals
-----------------
0   cat
1   dog
2   horse
3   donkey
4   pig

Answer 1

如果每行总是只有一个非缺失值，请使用前向填充缺失值(如 DataFrame.fillna 和 method='ffill' )，然后按位置选择最后一列 DataFrame.iloc , 也适用于一列 DataFrame添加 Series.to_frame :

df = df.ffill(axis=1).iloc[:, -1].to_frame('new')
print (df)
      new
0     cat
1     dog
2   horse
3  donkey
4     pig

如果可能，每行使用更多非缺失值 DataFrame.stack 与 join每个第一级:

print (df)
    p1   p2     p3      p4    p5
0  cat  NaN    NaN     NaN  lion
1  NaN  dog    NaN     NaN   NaN
2  NaN  NaN  horse     NaN   NaN
3  NaN  NaN    NaN  donkey   NaN
4  NaN  NaN    NaN     NaN   pig

df2 = df.stack().groupby(level=0).apply(', '.join).to_frame('new')
print (df2)
         new
0  cat, lion
1        dog
2      horse
3     donkey
4        pig

或 lambda 函数:

df2 = df.apply(lambda x: x.dropna().str.cat(sep=', '), axis=1).to_frame('new')
print (df2)
         new
0  cat, lion
1        dog
2      horse
3     donkey
4        pig