2015-09-04 37 views
0

使用astype時出現問題,並將值乘以255. 問題是,此CSV的第一列不應該由asType更改,或者因爲它爲類的標籤而倍增。Python,在陣列上部分使用Astype

我怎樣才能做這個轉換到'uint8'和乘255 255而不改變csv的0 col?

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') 

回答

0

我解決了這樣:

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())