Вы можете найти numpy.loadtxt полезным.
Например, предположим, у вас есть файл с таким содержанием:
Файл данных:
(id) 0 1
0 1 smooth
1 11 choppy
2 20 turbulent
3 2 smooth
4 5 choppy
5 7 bumpy
Затем вы можете загрузить данные в структурированный массив с помощью
import numpy as np
arr=np.loadtxt('datafile',
dtype=[('id','int'),('speed','float'),('surface','|S20')],
skiprows=1)
Обратите внимание, что вы можете пропустить первую строку файла данных, указав skiprows=1
.
Затем вы можете обращаться к строкам, как обычно, с числовыми индексами, такими как arr[1]
,
и вы можете получить доступ к столбцам по именам, таким как arr['speed']
.
И вы можете получить скорость в 3-м ряду с помощью arr[3]['speed']
или arr['speed'][3]
.
Для получения дополнительной информации о структурированных массивах см.
http://docs.scipy.org/doc/numpy/user/basics.rec.html