python - 如何从numpy数组中提取元素

Question

ls_ord_symbols     = np.zeros((len(na_csv_orders), 1), dtype='S5')

>>> print ls_ord_symbols
[['AAPL']
 ['IBM']
 ['GOOG']]

>>> print type(ls_ord_symbols)
<type 'numpy.ndarray'>
>>> print type(ls_ord_symbols[0])
<type 'numpy.ndarray'>
>>> print ls_ord_symbols[0][0]
AAPL
>>> print type(ls_ord_symbols[0][0])
<type 'numpy.string_'>
>>> print str(ls_ord_symbols[0][0])
AAPL
>>> print type(str(ls_ord_symbols[0][0]))
<type 'str'>

问题> 我需要使用原始类型提取存储在 numpy.array 中的每个元素。在这里，ls_ord_symbols将字符串存储在 numpy.array 中。

我必须使用以下方式来提取原始元素：

str(ls_ord_symbols[i][0])

给定索引，有没有更好的方法来做到这一点i？基本上，我希望简单地获取 ['AAPL', 'IBM', 'GOOG'] 或 'AAPL', 'IBM', 'GOOG' 的列表作为每个单独的字符串，同时使用索引遍历这个 numpy.arrayi

Answer 1

#!/usr/bin/env python
import numpy as np
ls_ord_symbols = np.zeros((3, 1), dtype='S5')
ls_ord_symbols[0,0] = 'AAPL'
ls_ord_symbols[1,0] = 'IBM'
ls_ord_symbols[2,0] = 'GOOG'
mylist = [ls_ord_symbols[i,0]
          for i in xrange(ls_ord_symbols.shape[0])]
print repr(mylist)