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]])

    得到的結果如下：