2

如何从 numpy ndarray 中的每个维度获取第一个元素?

import numpy
A = numpy.array([['a','b','c'],['d','e','f'],['g','h','i']])

结果应该是:

Result = ['a','d','g']
4

3 回答 3

4
>>> import numpy
>>> A = numpy.array([['a','b','c'],['d','e','f'],['g','h','i']])
>>> A[:,0]
array(['a', 'd', 'g'],
      dtype='|S1')
>>> A[...,0]
array(['a', 'd', 'g'],
      dtype='|S1')

请参阅索引(基本) - NumPy 手册索引 - NumPy 手册

于 2013-11-11T15:40:18.530 回答
2

你试过这个吗?

  list( A[:,0] )

索引numpy 数组通常会返回另一个 Numpy 数组,因此如果需要列表,则需要列表构造函数

于 2013-11-11T15:40:26.577 回答
1

使用 take 函数

import numpy
A = numpy.array([['a','b','c'],['d','e','f'],['g','h','i']])
print A.take((0,), 1)
于 2013-11-11T15:42:29.603 回答