python划分训练集和测试集_用python将一个数据文件分为训练集和测试集

#【导入相应的库（对数据库进行切分需要用到的库是sklearn.model_selection 中的 train_test_split）】

import numpy as np

from sklearn.model_selection import train_test_split

#【首先，读取.CSV文件成矩阵的形式。】

#若标签为浮点，直接使用下面一行即可

#data = np.loadtxt(open("iris.csv"),delimiter=",",skiprows=0)

##--------------------【若标签为Striing,先将标签转化为浮点型】------------------------------

def iris_type(s):

class_label={b'Iris-setosa':0,b'Iris-versicolor':1,b'Iris-virginica':2}

return class_label[s]

#使用numpy中的loadtxt读入数据文件（csv格式的iris数据，也可直接换成txt格式）

filepath='iris_data.csv' # 数据文件路径

data=np.loadtxt(filepath,dtype=float,delimiter=',',converters={4:iris_type})

##-------------------------------------------------------------------------------------

#【将矩阵最后一列之前的数值给X（输入数据），将矩阵最后一列的数值给yÿ