python中二维数组切片操作,一次操作取任意行任意列
时间: 2023-12-18 08:04:04 浏览: 145
1.6 编程基础之一维数组 python版.zip
5星 · 资源好评率100%
在Python中,可以使用numpy库来操作多维数组。numpy提供了更加方便和高效的多维数组切片操作。
以下是使用numpy库实现一次操作取任意行、列的示例代码:
```
import numpy as np
my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 取第一行和第三行,第二列和第三列
sub_array = my_array[[0, 2]][:, [1, 2]]
print(sub_array)
```
在这个例子中,`my_array`是一个3x3的二维数组,通过使用numpy库,可以使用一次操作来取得任意行、列的子数组。在`sub_array`中,使用双重索引分别取得第一行和第三行,然后使用逗号分隔符分别取得第二列和第三列,从而得到了一个2x2的子数组。需要注意的是,numpy库的多维数组切片操作使用的是方括号而不是圆括号。
阅读全文