X数组为坐标点,如下图所示(shape为500*2)
在这里提取admitted的时候,通过argwhere来选取y(label)符合条件的索引号,得到的是二维数组,如下图,argwhere(y==1)为(shape为127*1)
那么用np.argwhere(y==1)作为X的索引,假设A = np.argwhere(y==1),即X[A]输出为三维的矩阵(shape为127*1*2),结果如下图
分析以上过程:我一开始的想法是把X[A]中的A直接替换成A的array&
X数组为坐标点,如下图所示(shape为500*2)
在这里提取admitted的时候,通过argwhere来选取y(label)符合条件的索引号,得到的是二维数组,如下图,argwhere(y==1)为(shape为127*1)
那么用np.argwhere(y==1)作为X的索引,假设A = np.argwhere(y==1),即X[A]输出为三维的矩阵(shape为127*1*2),结果如下图
分析以上过程:我一开始的想法是把X[A]中的A直接替换成A的array&