初始化一个numpy array,形状是(3,4;在第0轴获得极大值,并打印结果和形状;在第1轴获得极大值,保持维度,并打印结果和形状
时间: 2024-09-25 15:13:27 浏览: 29
要使用NumPy创建一个数组并操作其最大值,你可以按照以下步骤进行:
首先,导入NumPy库:
```python
import numpy as np
```
然后,定义一个形状为(3, 4)的二维数组,通常你可以随机生成数据:
```python
data = np.random.rand(3, 4)
```
接下来,获取数组在第一个轴(0轴)上的最大值,这将返回一个大小为(4,)的一维数组:
```python
max_0_axis = data.max(axis=0)
print("Max along the first axis (axis=0):", max_0_axis)
print("Shape after maximizing along axis 0:", max_0_axis.shape)
```
接着,在第二个轴(1轴)上获取最大值,保留原始维度((3, 1)):
```python
max_1_axis = data.max(axis=1).reshape(-1, 1)
print("Max along the second axis (axis=1), keeping dimensions:", max_1_axis)
print("Shape after maximizing along axis 1:", max_1_axis.shape)
```
这里,`reshape(-1, 1)`用于确保新数组有两列,即使原始一维数组只有一个元素。
阅读全文