python - 将元素值保留在矩阵中

我有一个带有标签和图像的文本数据集。标签是代表手写数字的一维元素。 Dimension:(1010,)。图片为28 * 28像素大小的图片。Dimension:(1010, 784)。从文本数据集中读取后，我得到以下数据集reformatData['data']和reformatData['target']-分别为[n_samples, n_features]和[n_samples]。

同样，这些尺寸：(1010, 784) (1010,)打印时reformatData

现在，我尝试进行二进制分类并将数字引入矩阵，并尝试使用以下功能。

digits1=[8]
digits2=[1]


def read(digits):
    rows=28
    cols=28
    #lbl = array("b", reformatData['target'])
    lbl = reformatData['target']
    img=reformatData['data']
    #img = array("B", reformatData['data'])

    ind = [ k for k in xrange(len(lbl)) if lbl[k] in digits]
    images =  matrix(0, (len(ind), rows*cols))
    labels = matrix(0, (len(ind), 1))
    for i in xrange(len(ind)):
        images[i, :] = img[ ind[i]*rows*cols : (ind[i]+1)*rows*cols ]
        labels[i] = lbl[ind[i]]
    return images, labels

print read(digits=digits1)

输出量

(<0x784 matrix, tc='i'>, <0x1 matrix, tc='i'>)

我期望：

(<1010x784 matrix, tc='i'>, <1010x1 matrix, tc='i'>)

我该如何解决？

最佳答案

使用numpy where进行矢量化和更快的计算：

使用rahfl23的数组：

np.where(s==6, 0, 1)

对于矩阵：

np.where(images==6, 0, 1)