对数据进行排序、掩膜提取

代码：

import numpy as np
a = np.array([2,4,0])
b = sorted(a)       # 给 a 从小到大进行排序
print("a sorted之后的值：\n",b)
c = np.array([[5,6,9,8,5,6,8],
              [5,1,9,0,5,6,8],
              [5,4,8,8,4,6,2],
              [5,1,8,0,4,0,2],
              [5,4,2,7,4,9,2]])
print("c 的维度：\n", c.shape)
d = np.zeros(5, dtype=np.bool) # 创建一个大小为 5 的全 0 数组
print("交换位置之前c：\n{}".format(c))
c[a] = c[b]  # 给 c 的数据排个序。将 [2,4,0] 行的数据进行提取，并放在 c 的 [0, 2, 4] 处。
print("交换位置之后c：\n{}".format(c))

d[a] = True  # 将 d 按照 a 的值(该值就是索引，这里a或b都是一样的)提取后赋值为 True，其余赋值为 False
print("d 的输出：\n", d)

y_train_mask = c[d]  # 按照 d 中为 Ture 的位置，对 c 进行提取。（提取的是第一维度）
print("输出 y_train_mask：\n", y_train_mask)

输出：

D:\Anaconda3\python.exe D:/0_me_python/1.py
a sorted之后的值：
 [0, 2, 4]
c 的维度：
 (5, 7)
交换位置之前c：
[[5 6 9 8 5 6 8]
 [5 1 9 0 5 6 8]
 [5 4 8 8 4 6 2]
 [5 1 8 0 4 0 2]
 [5 4 2 7 4 9 2]]
交换位置之后c：
[[5 4 2 7 4 9 2]
 [5 1 9 0 5 6 8]
 [5 6 9 8 5 6 8]
 [5 1 8 0 4 0 2]
 [5 4 8 8 4 6 2]]
d 的输出：
 [ True False  True False  True]
输出 y_train_mask：
 [[5 4 2 7 4 9 2]
 [5 6 9 8 5 6 8]
 [5 4 8 8 4 6 2]]

Process finished with exit code 0