How to find the average of arrays (an array column) on 0th axis in a PySpark dataframe?

Question

I have a PySpark dataframe-

df = spark.createDataFrame([
    ("u1", [[1., 2., 3.], [1., 2., 0.], [1., 0., 0.]]),
    ("u2", [[1., 10., 0.]