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

怎么理解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 R语言单细胞测序 对seurat对象进行标准化时出现问题
  • ¥15 Anaconda navigator updateor一直更新
  • ¥15 请问如何在matlab里使用raven工具?
  • ¥100 关于ios手游充值到一定金额,再点充值提示公众号的问题。
  • ¥15 求一个stm32串口控制程序
  • ¥20 Windows 驱动开发版本疑问相关
  • ¥15 MAC 未能打开磁盘映像
  • ¥15 fastcap使用,二维导体输入问题
  • ¥15 hosts修改后不能访问
  • ¥15 关于化学反应速率C++编译问题/FLUENT