有没有大神帮我看看这个怎么写。

Question

Answer 1

>>> import numpy as np
>>> data = np.array([
    [6, 9, 1],
    [1, 1, 0],
    [8, 7, 0],
    [5, 6, 0],
    [5, 3, 1],
    [8, 8, 0],
    [8, 1, 2],
    [8, 7, 0],
    [1, 2, 2],
    [9, 4, 2]
])
>>> X_train = data[:,:2]
>>> y_train = data[:,2]
>>> X_train
array([[6, 9],
       [1, 1],
       [8, 7],
       [5, 6],
       [5, 3],
       [8, 8],
       [8, 1],
       [8, 7],
       [1, 2],
       [9, 4]])
>>> y_train
array([1, 0, 0, 0, 1, 0, 2, 0, 2, 2])
>>> X_train[np.where(y_train==0)]
array([[1, 1],
       [8, 7],
       [5, 6],
       [8, 8],
       [8, 7]])
>>> X_train[np.where(y_train==1)]
array([[6, 9],
       [5, 3]])
>>> X_train[np.where(y_train==2)]
array([[8, 1],
       [1, 2],
       [9, 4]])