txt读数据
从简单数据集中读取数据:
源码:
1 import random 2 import numpy as np 3 import matplotlib.pyplot as plt 4 def DatasetLoader(path): 5 f = open(path,"r") 6 position = 90 7 X=[] 8 y=[] 9 f1 = f.readlines() 10 random.shuffle(f1) 11 for data in f1: 12 temp_data = data.split() #将数据分割开来,.strip()函数去除数据两侧的空格/空白(\t等) 13 X.append([float(temp_data[0]),float(temp_data[1])]) #从文本读出来的是字符串文本,需要换成数值格式,不然后续处理数据会报错 14 y.append(int(temp_data[-1])) 15 X_train = np.array(X[:position]) 16 y_train = np.array(y[:position]) 17 X_test = np.array(X[position:]) 18 y_test = np.array(y[position:]) 19 return X_train,y_train,X_test,y_test, X, y
原始数据:
函数返回: