python - 基于在 Pandas 中堆叠列来延长 DataFrame

Question

我正在寻找实现以下功能的功能。最好在示例中显示。考虑:

pd.DataFrame([ [1, 2, 3 ], [4, 5, np.nan ]], columns=['x', 'y1', 'y2'])

看起来像:

   x  y1   y2
0  1   2  3
1  4   5  NaN

我想折叠 y1 和 y2 列，在必要时加长 DataFame，以便输出为:

   x  y
0  1   2  
1  1   3  
2  4   5  

也就是说，x 和 y1 或 x 和 y2 之间的每个组合对应一行。我正在寻找一个相对有效地执行此操作的函数，因为我有多个 y 和许多行。

Answer 1

您可以使用 stack完成任务，即

pd.DataFrame(df.set_index('x').stack().reset_index(level=0).values,columns=['x','y'])

     x    y
0  1.0  2.0
1  1.0  3.0
2  4.0  5.0