Python Numpy数组使用列表索引

      用一个例子理解一下

# -*- encoding=utf-8 -*-
'''
    author : James-J
    time : 2019/05/29
'''
import numpy as np

if __name__ == '__main__':
    data = np.array([[1, 2, 3, 4],
                     [5, 6, 7, 8],
                     [9,10,11,12]])
    print(data)
    print('-----------------------------')
    # 下标第0行第2列   3
    print('data[0, 2]             ', data[0, 2])
    print('-----------------------------')
    # 下标第1行第3列   8
    print('data[1, 3]             ', data[1, 3])
    print('-----------------------------')
    # 选取第0、2行
    print('data[[0, 2], ]          \n', data[[0, 2], ])
    print('-----------------------------')
    # 选取第0、2列
    print('data[:, [0, 2]]        \n', data[:, [0, 2]])
    print('-----------------------------')
    # 第一个列表表示行数 第二个表示列数
    # 选取第0行的第2列   3
    # 选取第1行的第3列   8
    # 返回的是ndarray(不是list)   [3 8]
    print('data[[0, 1], [2, 3]]   ', data[[0, 1], [2, 3]])
    print('-----------------------------')
    # 还可以这么玩 多选几次重复的没关系
    print('data[[1, 2, 1, 0, 2]]   \n', data[[1, 2, 1, 0, 2]])

    得到的结果如下：