D222097 2019-10-21 19:27 采纳率: 40%
浏览 2685
已采纳

怎么理解X_train[y_train == i][0]????

是在处理mnist数据集时遇到的问题。

X_train, y_train = load_mnist('E:\数据集\MINIST', kind ='train')_
#打印出来的X_train的形状为(60000, 784)
#打印出来的y_train的形状为(60000,)

在下面这段代码中遇到了问题:

for i in range(10):
    img = X_train[y_train == i][0].reshape(28, 28)
    test = X_train[y_train == i][0]
#    print(y_train == i)#出来的结果是[true ,false...]
#    print((y_train == i).shape)#出来的结果是(6000,)
#    print(test.shape)#(784,)

怎么理解 X_train[y_train == i][0]这一部分呢?
X_train明明是二维数组,X_train[][]第一个[]内不应该是指定行,第二个[]不应该是指定列,出来的不应该是一个元素吗?

还有y_train == i出来的是6000个true/false值组成的,怎么理解?

  • 写回答

2条回答 默认 最新

  • 毕小宝 博客专家认证 2019-10-21 20:12
    关注

    楼主可以试试先打印 X_train[y_train == i] 这个,这是一个推导式,返回 X_train 中满足 y_train ==i 条件的记录。
    后面那个 [0] 操作则是获取索引位置为 0 的数据。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

悬赏问题

  • ¥20 Windows10系统命令行调用
  • ¥15 php环境如何实现国密SM2相关功能
  • ¥15 关于#单片机#的问题:K210 例程里面提示我iomem.h: No such file or directory
  • ¥15 LSPI算法的大问题
  • ¥15 java导出word 动态填充数据
  • ¥20 扑克的算牌公式及软件制作
  • ¥20 如何通过云图中RPG去计算云图上不同位置的值?
  • ¥15 请问不小心下载到了钓鱼软件怎么办?
  • ¥15 求国博抢票 求国博抢票 有的私
  • ¥50 swiftui @query 报错