使用变量将列从numpy数组中隔离出来?

时间:2015-12-11 19:44:24

标签: python arrays numpy slice

我正在尝试隔离numpy数组的最后一列。但是,该函数需要适用于不同大小的数组。当我这样说时:

array[:,array_length] 
#array_length is a variable set to the length of one row of the array

看起来它会工作,它返回一个错误,告诉我我不能用变量切片,但只能用整数。

有没有办法用我没看到的numpy做到这一点?

1 个答案:

答案 0 :(得分:1)

要访问numpy数组的最后一列,您可以使用-1

last_col = array[:, -1]

或者你也可以

array_length = len(array[0]) - 1

last_col = array[:, array_length]