python - 按特定顺序操作列表和数组

Question

我有一个表单列表:

['a b 1 2', 'c d 3 4']

我最终试图得到 4 个垂直 numpy 数组。例如，['a','c'] 和 [1,2]

我对各种分割函数、数组分割等的使用感到有点困惑。

super 菜鸟问题，这更像是一个尽可能高效地完成此操作的练习。

任何帮助将不胜感激!

Answer 1

您可以将其作为记录数组读入:

>>> A = ['a b 1 2', 'c d 3 4']
>>> from StringIO import StringIO
>>> import numpy
>>> s = StringIO('\n'.join(A))
>>> data = numpy.genfromtxt(s, dtype=[('letter1', 'S1'), ('letter2', 'S1'), ('num1', 'f8'), ('num2', 'f8')])

然后访问列:

>>> data['letter1']
array(['a', 'c'], 
      dtype='|S1')
>>> data['num1']
array([ 1.,  3.])

请注意，这仅限于固定大小的字符串。不确定这是否是您的数据的问题。