0
点赞
收藏
分享

微信扫一扫

[数据处理] 加载csv文件

火热如冰 2022-02-27 阅读 84

加载csv文件。

  1. 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)

参考:

  1. ML2021Spring - HW1.ipynb:https://colab.research.google.com/github/ga642381/ML2021-Spring/blob/main/HW01/HW01.ipynb#scrollTo=hsNO9nnXQBvP
举报

相关推荐

0 条评论