从数据框删除名称列

时间:2018-09-09 09:03:53

标签: python dataframe

我从iris.csv文件创建一个数据框:

raw_data = pd.read_csv(“ iris.csv”,标头= 0)

raw_data

Species     Sepal.Length    Sepal.Width     Petal.Length    Petal.Width
0   setosa  5.1     3.5     1.4     0.2
1   setosa  4.9     3.0     1.4     0.2
2   setosa  4.7     3.2     1.3     0.2
3   setosa  4.6     3.1     1.5     0.2
4   setosa  5.0     3.6     1.4     0.2
5   setosa  5.4     3.9     1.7     0.4
6   setosa  4.6     3.4     1.4     0.3
7   setosa  5.0     3.4     1.5     0.2
8   setosa  4.4     2.9     1.4     0.2

然后我提取数据框中的第一列;

train_y = raw_data.ix[:,0]

train_y

0         setosa
1         setosa
2         setosa
3         setosa
4         setosa
5         setosa
6         setosa
7         setosa
8         setosa
9         setosa
10        setosa

但是当我尝试提取其他列组成的train_X时:

train_X = raw_data.ix[:,1:]

我也得到列名 有什么方法可以从train_X删除名称列

谢谢

1 个答案:

答案 0 :(得分:0)

要获取数据,请尝试以下操作:

train_X = raw_data.iloc[:, 1:]