Numpy 二维数组的三大操作

zltzlt

07 - 二维数组必备的三大操作

单选

1. >>> b[3, 3]
   
2. 'a33'
   
3. >>> b[3][3]
   
4. 'a33'

复制代码

选整行：

1. >>> b[3]
   
2. array(['a30', 'a31', 'a32', 'a33'], dtype='<U3')
   
3. >>> b[3, :]
   
4. array(['a30', 'a31', 'a32', 'a33'], dtype='<U3')

复制代码

选整列：

1. >>> b[:, 3]
   
2. array(['a03', 'a13', 'a23', 'a33', 'a43', 'a53'], dtype='<U3')

复制代码

多选

选取 2 到 5 行：

1. >>> b[2: 5]
   
2. array([['a20', 'a21', 'a22', 'a23'],
   
3.        ['a30', 'a31', 'a32', 'a33'],
   
4.        ['a40', 'a41', 'a42', 'a43']], dtype='<U3')
   
5. >>> b[2: 5, :]
   
6. array([['a20', 'a21', 'a22', 'a23'],
   
7.        ['a30', 'a31', 'a32', 'a33'],
   
8.        ['a40', 'a41', 'a42', 'a43']], dtype='<U3')

复制代码

选取前 3 列：

1. >>> b[:, : 3]
   
2. array([['a00', 'a01', 'a02'],
   
3.        ['a10', 'a11', 'a12'],
   
4.        ['a20', 'a21', 'a22'],
   
5.        ['a30', 'a31', 'a32'],
   
6.        ['a40', 'a41', 'a42'],
   
7.        ['a50', 'a51', 'a52']], dtype='<U3')

复制代码

选取一部分：

1. >>> b[1: 3, : 5]
   
2. array([['a10', 'a11', 'a12', 'a13'],
   
3.        ['a20', 'a21', 'a22', 'a23']], dtype='<U3')

复制代码

还能指定步长：

1. >>> b[1: 6: 2, : 5]
   
2. array([['a10', 'a11', 'a12', 'a13'],
   
3.        ['a30', 'a31', 'a32', 'a33'],
   
4.        ['a50', 'a51', 'a52', 'a53']], dtype='<U3')

复制代码

花式索引

1. >>> b[np.ix_([1, 4, 5], [0, 3])]    # 选取 1、4、5 行的 0、3 列
   
2. array([['a10', 'a13'],
   
3.        ['a40', 'a43'],
   
4.        ['a50', 'a53']], dtype='<U3')

复制代码