在 sparkR 中附加 DataFrame 中的一个元素

Attach one element from a DataFrame in sparkR

我在 sparkR 中有一个名为 'data' 的 DataFrame。 'Data' 包含 'user'、'amount_spent' 和 'amount_won'。 我想为用户 1.

计算余额= amount_spent - amount_won
y <- filter(data, data$user==1)

现在我计算总和

yn <- agg(groupBy(y, "user"), amount_spent="sum", amount_won="sum")

现在我计算用户 1 的余额

ynn <- withColumn(yn, "balance", yn[[3]] - yn[[2]])

这一切都给了我一个正确的结果,但是我想附加 "balance" 并将其作为来自 'ynn' 的整数,这是一个 DataFrame。我怎样才能做到这一点?如果我想为 100 个用户执行此操作,我需要将相同的操作执行 100 次。

我可能遗漏了一些东西,但为什么不这样做:

## The data set is now `data` not yn
yn = agg(groupBy(data, "user"), amount_spent="sum", amount_won="sum")

现在计算余额时,每个用户都有余额

ynn = withColumn(yn, "balance", yn[[3]] - yn[[2]])