Python,在数组中部分使用Astype

2024-06-26 17:06:37 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个问题,使用astype和乘以255的值。 问题是,此CSV的第一列不应被asType更改或相乘,因为它是类的标签。你知道吗

在不更改csv的0列的情况下,如何将此转换为“uint8”并乘以255?你知道吗

  with open(trainingFolder + "standardization_traindata.csv", 'wb') as fp:
        for i in range(0, aux_data.shape[0]):
            column = aux_data[i, :].tolist()
            #
            column = np.array(column).astype('uint8') * 255
            #
            column = map(lambda x: str(x) + ',', column)
            column = ''.join(column)[0:-1]
            fp.write(column + '\n')

Tags: csvdatawith情况column标签openfp
1条回答
网友
1楼 · 发布于 2024-06-26 17:06:37

我解决了这个问题:

with open(testingFolder + "zca_whitening_testdata.csv", 'wb') as fp:
    writer = csv.writer(fp)

    temp_list = []

    temp_list.append("label")

    for j in range(0,aux_data.shape[1]-1):
        header = "pixel"+str(j)
        temp_list.append(header)

    writer.writerow(temp_list)

    for i in range(0, aux_data.shape[0]):
        writer.writerow(aux_data[i,:].tolist())

相关问题 更多 >