python读取csv和txt数据转换成向量

最近写程序需要从文件中读取数据,并把读取的数据转换成向量。

查阅资料之后找到了读取csv文件和txt文件两种方式,下面结合自己的实验过程,做简要记录,供大家参考:

1、读取csv文件的数据

import csv

filtpath = "data_test.csv"

with open(filtpath,'r') as csvfile:
    reader = csv.reader(csvfile)
    header = next(reader)
    data = []
    for line in reader:
        data.append(line)
    print(header)
    print(data) 

使用上面的代码读取数据之后,看似是向量格式,可是数据类型却是字符串类型,不符合要求。

2、读取txt文件

import numpy as np

#load data
file=open("data.txt")
lines=file.readlines()
rows=len(lines)

datamat = np.zeros((rows,15))

row=0
for line in lines:
    line=line.strip().split('\t')
    datamat[row,:]=line[::]
    row+=1

print(datamat)
print(datamat.shape) 

使用读取txt的方式可以使读取的数据既是向量格式,数据的数据类型也没有发生变化


注:上面的程序都出自下面的参考链接

参考:

1.python读取csv文件

2.python读取txt文件

猜你喜欢

转载自blog.csdn.net/jp_666/article/details/79853046