txt读数据

prince-wzm / 2023-05-03 / 原文

从简单数据集中读取数据:

源码:

 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

 

原始数据:

函数返回: