当前位置:主页 > 软件编程 > Python代码 >

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

时间:2021-07-01 08:52:22 | 栏目:Python代码 | 点击:

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

查阅资料之后找到了读取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的方式可以使读取的数据既是向量格式,数据的数据类型也没有发生变化

您可能感兴趣的文章:

相关文章