选择第n列X = np.array(data.ix [:,...]

时间:2017-05-22 16:17:20

标签: numpy

以下代码用于numpy数组的前四列中的数据。

X=np.array(data.ix[:,0:4])

如果我想要第1名,第7名和第8名怎么办?

2 个答案:

答案 0 :(得分:1)

只需使用列索引列表:

In [9]: ix=np.arange(20).reshape(2,10)

In [10]: ix
Out[10]: 
array([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14, 15, 16, 17, 18, 19]])

In [11]: ix[:,[0,6,7]]
Out[11]: 
array([[ 0,  6,  7],
       [10, 16, 17]])

答案 1 :(得分:1)

您似乎正在尝试将Dataframe切片为numpy数组。如果是这种情况,您可以使用:

x=data.iloc[:,[0,6,7]].values

ix将来会被弃用,所以请改用iloc。