1. 程式人生 > >python讀取csv和txt資料轉換成向量

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的方式可以使讀取的資料既是向量格式,資料的資料型別也沒有發生變化

注:上面的程式都出自下面的參考連結

參考: