python - 保留矩阵中的元素值

Question

我有一个带有标签和图像的文本数据集。标签是表示手写数字的一维元素。 尺寸:(1010，)。图片为 28*28 像素大小的图片。尺寸:(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'>)

如何解决这个问题？

Answer 1

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

使用上面的 rahfl23 数组:

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

对于矩阵:

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