我想学习一种使用Numpy在Python中快速生成2D数组的方法。我的当前代码使用生成器:points= [[x-w/2,y-h/2,img[y,x]] for y in range(h) for x in range(w) ifimg[y,x]!=0]注意: img是2D numpy.array,其大小为10×10。上面的代码本身并不慢,
我需要将这个Tensorflow数据集转换成两个NumPy数组,X_test包含输入,y_test包含标签,按的方式排序。([y for _, y in test_dataset], axis=0)
在同一个数据集中重复两次似乎是浪费的。我怀疑数据集是在我浏览过一次之后被洗牌的。然而,这也意味着,在我上面的片段中,X_test和y_test也不会以相同的方式排序。我怎么才能解决呢?tfds.as_numpy</em