Run
import numpy as np #initialize array A = np.array([[2, 1], [5, 4]]) #compute mean output = np.mean(A, axis=0) print(output)
Output