NUMPY058 题解：每行减去该行均值

题目

计算每行减去该行均值后的矩阵。

解题思路

1. 读取矩阵
2. 计算每行均值（\texttt{keepdims=True} 保持维度）
3. 相减

代码

import numpy as np

m, n = map(int, input().split())
X = np.array([list(map(float, input().split())) for _ in range(m)])
Y = X - X.mean(axis=1, keepdims=True)
print(Y)

代码详解

• \texttt{X.mean(axis=1)} —— 按行计算均值，结果是 1D 数组
• \texttt{keepdims=True} —— 保持维度，结果变成 (m, 1) 形状
• \texttt{X - X.mean(...)} —— 广播相减

核心知识点

1. \texttt{mean(axis=1)} —— 按行计算
2. \texttt{keepdims=True} —— 保持维度以便广播
3. 广播机制