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

Question

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

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

我必须做同样的事情，但是在pyspark。