编写一个函数,实现将numpy矩阵的每一行元素减去该行平均值
时间: 2023-05-18 07:06:25 浏览: 264
可以使用numpy库中的mean函数来计算每一行的平均值,然后使用广播机制将平均值减去每个元素,最后返回处理后的矩阵。以下是示例代码:
```python
import numpy as np
def subtract_row_mean(matrix):
row_means = np.mean(matrix, axis=1, keepdims=True)
return matrix - row_means
# 示例用法
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
result = subtract_row_mean(matrix)
print(result)
```
输出结果为:
```
[[-1. 0. 1.]
[-1. 0. 1.]
[-1. 0. 1.]]
```
阅读全文