python - 如何切片具有未知列数的numpy数组？

Question

我想将 2d numpy 数组切片为我选择的索引。例如，我有这个one.csv

111,222,333,444
1,2,3,4

我想把它切成例如

XX = array([[ 111.,  222.,  333.],
            [   1.,    2.,    3.]])

YY = array([ 444.,    4.])

假设我不知道有多少索引，所以XX我YY做到了

import numpy

data = numpy.loadtxt('one.csv', delimiter=",")


XX = data[:,0:len(data)]
YY = data[:,len(data)]

两者XX和YY结果如下

# XX
array([[ 111.,  222.],
       [   1.,    2.]])

# YY
array([ 333.,    3.])

我该如何解决这个问题？除了使用还有其他选择len(data)吗？

谢谢您的帮助。

Answer 1

对于您的具体示例，请使用以下命令：

# all rows, and all columns but the last
XX = data[:, 0:-1]

# all rows and only the last column
YY = data[:, -1]

data但总的来说，您可以根据需要获取数组的形状data.shape并切片列。