從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

使用匯整功能

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