根据标签数组y寻找数组X的对应位置
X和y都是数组,可以通过y的值找到对应位置的X的值。
x= np.array([[ 2.58819045e-01, 9.65925826e-01],
[-1.30526192e-01, 9.91444861e-01],
[ 2.06646660e-01, -1.08761429e-01],
[ 1.92387953e+00, 1.17316568e-01],
[-7.07106781e-01, 7.07106781e-01]])
y = np.array([0,0,1,0,1])
找到所有y=1对应的x的坐标:
x[y==1,0]
输出结果如下:
array([ 0.20664666, -0.70710678])