calculate column mean of a matrix - how to optimize?

Question

input:

[
[1,2,3,4,5],
[5,4,3,2,1],
[3,3,3,3,3]
]

output:

[3,3,3,3,3]

brute force solution:

def