python - 如何在python中以相同的顺序获取各种属性

Question

我有一个线条文件，这反过来又保存了每条线条的信息、速度、时间和表面类型。我想要做的是按照下面所示的顺序对 np.array 中的信息进行排序，其中 id 是行号。

(id)   0   1   2   3   4   5   6   7   8   9

0   t1 t2 t3 t4 t5 t6  t7 t8 t9 t10

1   t1 t2 t3 t4 t5 t6  t7 t8 t9 t10 

2   t1 t2 t3 t4 t5 t6  t7 t8 t9 t10 

3   t1 t2 t3 t4 t5 t6  t7 t8 t9 t10 

4   t1 t2 t3 t4 t5 t6  t7 t8 t9 t10 

5  t1 t2 t3 t4 t5 t6  t7 t8 t9 t10 

...感谢您的回复

Answer 1

您可能会发现numpy.loadtxt有用。

例如，假设您有一个包含以下内容的文件:

数据文件:

(id)   0   1   
0   1 smooth 
1   11  choppy
2   20  turbulent
3   2  smooth
4   5  choppy
5  7   bumpy

然后您可以使用以下命令将数据加载到 numpy 结构化数组中

import numpy as np
arr=np.loadtxt('datafile',
               dtype=[('id','int'),('speed','float'),('surface','|S20')], 
               skiprows=1)

请注意，您可以通过指定 skiprows=1 跳过数据文件的第一行.

然后您可以像往常一样使用数字索引访问行，例如 arr[1] ,您可以按名称访问列，例如 arr['speed'] .

你可以通过 arr[3]['speed'] 获得第三行的速度或arr['speed'][3] .

有关结构化数组的更多信息，请参阅 http://docs.scipy.org/doc/numpy/user/basics.rec.html