加载csv文件。
- csv.reader()
【数据读取->list,numpy->tensor】
# 数据处理需要的包
import csv
import numpy as np
with open(PATH, 'r') as f:
data = list(csv.reader(f)) # 使用csv.reader将文件读取进来,并转为list形式,方便后续处理
data = np.array(data[1:])[:, 1:].astype(float) # csv文件第一行常常为标题,第一列常常为id,可以分情况去掉不需要的数值。
csv.reader()
读取csv文件。
np.array
转为numpy形数组形式具有行、列的概念后更易处理。
.astype(float)
并统一为浮点数形式
处理好后,可以再将数据统一为tensor形式:
data = torch.FloatTensor(data)
参考:
- ML2021Spring - HW1.ipynb:https://colab.research.google.com/github/ga642381/ML2021-Spring/blob/main/HW01/HW01.ipynb#scrollTo=hsNO9nnXQBvP