numpy ---数组的索引与切片（2）

壹、二维数组

• 引入numpy模块，并定义一个二维数组arr1

import numpy as np
arr1= np.array([[1,2,3],[4,5,6],[7,8,9]])

• arr[:x] 获取二维数组的前x行数据
  例：获取数组arr1的前两行数据

print("原数据：",arr1)
print("切片数据：",arr1[:2])

结果

原数据： [[1 2 3]
 [4 5 6]
 [7 8 9]]
切片数据： [[1 2 3]
 [4 5 6]]

• arr[x,:y] 获取数组第x行前y列的数据
  例： 获取数组第二行前两列的数据

print("*************")
print("原数据：",arr1)
print("切片数据：",arr1[1,:2])

结果

原数据： [[1 2 3]
 [4 5 6]
 [7 8 9]]
切片数据： [4 5]

• arr1[:x,y] 获取数组第x列前y行的数据
  例：获取数组第三列前两行的数据

print("*************")
print("原数据：",arr1)
print("切片数据：",arr1[:2,2])

结果

*************
原数据： [[1 2 3]
 [4 5 6]
 [7 8 9]]
切片数据： [3 6]

• arr1[:,:x] 获取二维数组中所有数组的第x列数据（返回值为二维数组形式）
  例：获取二维数组中所有数组的第一列数据

print("*************")
print("原数据：",arr1)
print("切片数据：",arr1[:,:1