一些犯过的愚愚蠢蠢的错误(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)]
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_43368987/article/details/130129399