考虑到:
test = numpy.array([[1, 2], [3, 4], [5, 6]])
Test [i]给出第i行(例如[1,2])。如何访问第I列?(例如[1,3,5])。还有,这手术会很贵吗?
考虑到:
test = numpy.array([[1, 2], [3, 4], [5, 6]])
Test [i]给出第i行(例如[1,2])。如何访问第I列?(例如[1,3,5])。还有,这手术会很贵吗?
当前回答
如果你想一次访问多个列,你可以这样做:
>>> test = np.arange(9).reshape((3,3))
>>> test
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>> test[:,[0,2]]
array([[0, 2],
[3, 5],
[6, 8]])
其他回答
如果你想一次访问多个列,你可以这样做:
>>> test = np.arange(9).reshape((3,3))
>>> test
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>> test[:,[0,2]]
array([[0, 2],
[3, 5],
[6, 8]])
访问列0:
>>> test[:, 0]
array([1, 3, 5])
访问第0行:
>>> test[0, :]
array([1, 2])
这在NumPy引用的第1.4节(索引)中介绍。这很快,至少以我的经验来看。这当然比在循环中访问每个元素要快得多。
>>> test
array([[0, 1, 2, 3, 4],
[5, 6, 7, 8, 9]])
>>> ncol = test.shape[1]
>>> ncol
5L
然后你可以这样选择第2 - 4列:
>>> test[0:, 1:(ncol - 1)]
array([[1, 2, 3],
[6, 7, 8]])
这不是多维的。它是二维数组。您想要访问所需列的位置。
test = numpy.array([[1, 2], [3, 4], [5, 6]])
test[:, a:b] # you can provide index in place of a and b
要获得几个独立的列,只需:
> test[:,[0,2]]
你会得到第0列和第2列