将数据集按特征|列分割为解释变量 X & 响应变量 Y 的几种方法

波士顿房价预测 特点：回归问题，解释变量唯一 利用整数下标

代码语言：javascript复制

from pandas import read_csv

dataset =read_csv('train.csv').values
X = dataset[:,0:13]
Y = dataset[:,13]

波士顿房价预测 特点：回归问题，解释变量唯一 利用条件

代码语言：javascript复制

from pandas import read_csv

dataset =read_csv('train.csv')
X = X = dataset.iloc[:, dataset.columns != "price"]
Y = dataset[:,dataset.columns == "price"]

船舶航迹预测 特点：回归问题，解释变量为 lat lon

代码语言：javascript复制

from pandas import read_csv

dataset =read_csv('train.csv')
# mmsi	lat	lon	Sog	Cog	timestamp
#dataset.iloc[行,列]
#这里指  [所有行，bool]     
X = dataset.iloc[:, [True,False,False,True,True,True]]
Y = dataset.iloc[:, [False,True,True,False,False,False]]


#Tip：
#这里的列，根据bool/条件语句/整数去选择列都可以，比如
X = dataset.iloc[:, dataset.columns != "lat"]
#上面的只适合一元响应变量的特征输入，很可惜 携程下面这样就无法通过编译了
X = dataset.iloc[:, dataset.columns != "lat" and  dataset.columns != "lon"]
#原因如下 上面提到的双条件判断出现了[True,False,False,True,True,True]与[False,True,True,False,False,False]判断，出现了多组值的判断

csv dataset 变量 编译 数据

0 人点赞