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的形状，并根据需要对列进行切片。