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

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

用numpy的比较运算：

import numpy as np

arr=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]])
for i in range(3):
    print(f"分类为{i}的样本:\n", arr[arr[:, 2] == i][:, :2])

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632

非常感谢您使用有问必答服务，为了后续更快速的帮您解决问题，现诚邀您参与有问必答体验反馈。您的建议将会运用到我们的产品优化中，希望能得到您的支持与协助！

速戳参与调研>>>https://t.csdnimg.cn/Kf0y