numpy数组取一列并且维度变成2
时间: 2023-05-20 10:01:39 浏览: 54
可以使用numpy的切片操作来取一列,并使用reshape函数将维度变成2,代码如下:
```python
import numpy as np
# 创建一个3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 取第二列并将维度变成2
col = arr[:, 1].reshape(-1, 1)
print(col)
```
输出结果为:
```
array([[2],
[5],
[8]])
```
相关问题
numpy数组增加维度
要将numpy数组增加一个维度,你可以使用numpy的`expand_dims`函数。它可以在指定的位置插入一个新的维度。
以下是使用`expand_dims`函数将一维数组转换为二维数组的示例:
```python
import numpy as np
# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5])
# 使用expand_dims函数在第二个维度插入一个新维度
new_arr = np.expand_dims(arr, axis=1)
print(new_arr.shape) # 输出:(5, 1)
```
在这个例子中,我们在原始数组的第二个维度插入了一个新的维度,从而将一维数组转换为了一个列向量的二维数组。`axis=1`参数表示插入的位置是第二个维度。
你也可以在其他位置插入新的维度,只需要根据需要调整`axis`参数的值即可。
python求数组某一列的维度
假设你有一个二维数组 `arr`,要求获取其第二列的维度,可以使用以下代码:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col_dimension = arr[:, 1].shape
print(col_dimension)
```
输出结果为 `(3,)`,表示第二列有 3 个元素。