python - 如何有效地将多个 pandas 列组合成一个类似数组的列？

Question

创建(或加载)带有类似对象类型列的 DataFrame 很容易，如下所示:

[In]: pdf = pd.DataFrame({
                     "a": [1, 2, 3], 
                     "b": [4, 5, 6], 
                     "c": [7, 8, 9], 
                     "combined": [[1, 4, 7], [2, 5, 8], [3, 6, 9]]}
      )

[Out]
   a  b  c   combined
0  1  4  7  [1, 4, 7]
1  2  5  8  [2, 5, 8]
2  3  6  9  [3, 6, 9]

我目前所处的位置是，作为单独的列，我需要将值作为单个列返回，并且需要非常有效地这样做。有没有一种快速有效的方法可以将列组合成单个对象类型的列？

在上面的示例中，这意味着已经有列 a、b 和 c，我希望创建 合并。

我未能在网上找到类似的问题示例，如果这是重复的，请随时链接。

Answer 1

使用DataFrame.agg 并将列表作为聚合方法传递，使用axis=1，然后将其分配给新列

>>> pdf.assign(combined=pdf.agg(list, axis=1))

   a  b  c   combined
0  1  4  7  [1, 4, 7]
1  2  5  8  [2, 5, 8]
2  3  6  9  [3, 6, 9]