python数组取某一列
时间: 2024-05-08 14:14:10 浏览: 189
Python数组的使用
在Python中,我们可以使用numpy库中的数组对象来实现数组取某一列的操作。具体来说,可以使用数组对象的切片操作来实现。下面是一个简单的示例代码,假设我们有一个2行3列的数组a,我们想要取出第二列的数据:
```
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
col2 = a[:, 1]
print(col2)
```
输出结果为:
```
[2 5]
```
其中,`a[:, 1]`表示取出数组a中的所有行(用冒号表示),第二列数据(下标从0开始,所以是1)。因此,`col2`即为取出的第二列数据。
阅读全文