在 dataframe-js 中添加所有行值的最佳方法是什么？

Question

我有一个如下所示的数据框，我想将所有值添加到一行中并将总和存储在新列中。

var df = new dfjs.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6]}, ['col1', 'col2'])

所以我有：

col1 col2
1     4
2     5
3     6

我想得到：

col1 col2 col3
1    4    5
2    5    7
3    6    9

鉴于数据框包含很多行，最有效的方法是什么？

Answer 1

我发现我可以这样做：

var df = df.withColumn('col3', row => row.toArray().reduce((a, b) => a + b, 0));

为了安全起见，我将 NaN 值扩展为：

var df = df.withColumn('col3', row => row.toArray().reduce((a, b) => a + (isNaN(b) ? 0 : b), 0));