import numpy as np matrix = np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]) center = matrix.mean(axis=0, keepdims=True) centered = matrix - center print(center.shape) print(centered.sum(axis=0).tolist())