python - 从 numpy ndarray 中的每个维度获取特定元素

Question

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

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

结果应该是：

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

Answer 1

>>> 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 手册。

Answer 2

你试过这个吗？

  list( A[:,0] )

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

Answer 3

使用 take 函数

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