Numpy 二维数组的三大操作

zltzlt

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

单选

>>> b[3, 3]
'a33'
>>> b[3][3]
'a33'

选整行：

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

选整列：

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

多选

选取 2 到 5 行：

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

选取前 3 列：

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

选取一部分：

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

还能指定步长：

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

花式索引

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