pyspark - 如何在pyspark中按行计算列的平均值？

Question

我有一个包含 4 个数值列的数据框，我必须计算这些列的平均值并将平均值存储在 pyspark 的另一列中。

df["mean"] = df.loc[:,d_cols].apply(np.mean, axis=1) (python pandas)

我必须做与上面相同的事情，但在 pyspark 中。