python - 附加一个 numpy 数组，但以二维方式组织附加信息

Question

这感觉好像应该非常简单，但我很挣扎。

我有一些代码只读取文本文件。该文本文件非常复杂，但基本上包含三个旋转角度，看起来像这样:

HeaderInformation

datalocation    sometext     1    3.45   15.5    -120
datalocation2    sometext2     1    5.4   125    -140

所以我目前有代码:

data = textfile.readlines
variable = 'datalocation'
Rotations = np.array([])
for line in data:
    if variable in line:
        linesplit = line.split()
        AngleRot = linesplit[3]
        AngleTilt = linesplit[4]
        AnglePsi = linesplit[5]
        Rotations = np.append(Rotations, ([AngleRot], [AngleTilt], [AnglePsi]))

print RelionRotations

这可行，但所有角度都只是一个接一个地添加到 numpy 数组中。

我稍后需要使用这些角度进行计算，但我希望将它们分开，以便我可以获取文本文件中每行的旋转。即我希望它们在 2D numpy 数组中每行索引为 3 个值。

这看起来应该很容易，但我现在正在努力弄清楚如何做到这一点。

Answer 1

使用 numpy 自己的 csv 阅读器

In [29]: txt = '''HeaderInformation
    ...: 
    ...: datalocation    sometext     1    3.45   15.5    -120
    ...: datalocation2    sometext2     1    5.4   125    -140'''

In [32]: data = np.genfromtxt(txt.splitlines(),skip_header=2)
In [33]: data
Out[33]: 
array([[    nan,     nan,    1.  ,    3.45,   15.5 , -120.  ],
       [    nan,     nan,    1.  ,    5.4 ,  125.  , -140.  ]])

或者跳过 2 个文本列:

In [34]: data = np.genfromtxt(txt.splitlines(),skip_header=2,usecols=[2,3,4,5])
In [35]: data
Out[35]: 
array([[   1.  ,    3.45,   15.5 , -120.  ],
       [   1.  ,    5.4 ,  125.  , -140.  ]])