python nompy取某一列
时间: 2023-06-29 22:17:15 浏览: 175
Python numpy 提取矩阵的某一行或某一列的实例
在NumPy中,可以使用切片(slice)来获取数组的某一列。假设有一个二维数组 `arr`,要获取其中的第二列,可以使用如下代码:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col2 = arr[:, 1]
print(col2)
```
其中 `:` 表示取全部行,`1` 表示取第二列(从0开始计数)。上述代码输出的结果为:
```
[2 5 8]
```
这是一个一维数组,包含了原数组中的第二列元素。需要注意的是,获取的列是一个视图(view)而不是副本(copy),对视图所做的修改会反映到原数组中。如果需要获取副本而不是视图,可以使用 `copy()` 方法,例如:
```python
col2_copy = arr[:, 1].copy()
```
这样就可以获取一个副本了,对副本的修改不会影响原数组。
阅读全文