python+numpy按行求一个二维数组的最大值方法,给出代码
时间: 2024-02-04 18:04:09 浏览: 70
python+numpy按行求一个二维数组的最大值方法
可以使用numpy库中的amax函数来求二维数组的最大值,可以通过axis参数指定是按行还是按列进行求解。以下是按行求解的代码示例:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_values = np.amax(arr, axis=1)
print(max_values) # 输出 [3 6 9]
```
在上述代码中,我们定义了一个3x3的二维数组arr,并使用np.amax函数按行求解最大值,最后输出结果。如果要按列求解最大值,只需要将axis参数设置为0即可。
阅读全文