一些犯过的愚愚蠢蠢的错误（1）

一些犯过的愚愚蠢蠢的错误（1）

背景介绍

统计每个label的index存在digit_indices中，如果有5个label，那么digit_indices应该是有五个list元素的list

• data 是一个list
• label是与其对应的list(0,1,2,3.)

Bug

digit_indices = [np.where(label == i)[0] for i in range(num_classes)]

没有任何error，只是在运行过程中发现digit_indices是空list，诡异。

探索中

看上去都是int为什么label==i 的判断后的list是空？（黑人问号），这个时候我还是没有想到int32和int64的区别。

• 只是尝试写成：
  digit_indices = [np.where(label == i)[0] for i in np.unique(label)]
  

这次的digit_indices不是空的了，但是为什么呢？

• 改成这样试试：
  digit_indices = [np.where(label == i)[0] for i in [0,1]]
  
  空的
• 灵光了，意识到是type的问题，label的int是int64，但是for循环中的i是int32，等不了
  digit_indices = [np.where(label == i)[0] for i in np.array([0,1],dtype=int64)]
  
  哈哈哈，被自己刚睡醒的愚蠢笑掉大牙，dtype哪里有int64的说法？？

上正经的！

digit_indices = [np.where(label == i)[0] for i in np.array([0,1],dtype=int)]