写一个程序实现找到二维数组np.arange(9).reshape(3,3)每一行中的最大值?
时间: 2024-05-07 12:16:29 浏览: 106
python+numpy按行求一个二维数组的最大值方法
可以使用numpy库的amax函数来实现,具体代码如下:
```python
import numpy as np
arr = np.arange(9).reshape(3,3)
max_vals = np.amax(arr, axis=1)
print(max_vals)
```
输出结果为:
```
[2 5 8]
```
其中,axis=1表示在每一行中查找最大值。
阅读全文