numpy中结构数组

   在c语言中,咱们能够使用关键字struct定义结构类型。和c语言同样,numpy也能够建立结构定义,这样能够很方便的读取二进制的C语言结构数组,将其转换为numpy数组对象,假设咱们定义的结构数组以下(C语言描述):python

struct Person{
     char name[30];
     int    age;
     float weight;  
};

咱们在python中能够自定义类型以下:数组

>>> persontype = np.dtype({
    'names':['name','age','weight'],
    'formats':['S30','i','f']},align = True)

>>> a = np.array([('Liming',24,63.9),('Mike',15,67.),('Jan',34,45.8)],dtype = persontype)
>>> a
array([('Liming', 24, 63.900001525878906), ('Mike', 15, 67.0),
('Jan', 34, 45.79999923706055)],
dtype={'names':['name','age','weight'], 'formats':['S30','<i4','<f4'], 'offsets':[0,32,36], 'itemsize':40, 'aligned':True})orm

首先建立了dtype对象persontype,它的结构类型是一个藐视各字段的字典,该字典有两个键:‘names’和'format'.每一个键对应的值是一个列表,‘names’定义结构体中每一个字段的名称,'formats’定义每一个字段的数据类型。咱们使用的类型以下:对象

‘S30’:表示长度为30的字符串类型,因为结构体中每一个元素的大小固定,因此须要指定长度。blog

‘i’:32位的整数类型,至关于np.int32字符串

'f':32位单精度浮点数,至关于np.float32string

而后经过np.array()建立数组,经过dtype指定元素类型为persontypeit

还能够使用包含该多个元组的列表来描述数据类型,刑如:(字段名,数据类型)form

>> persontype = np.dtype([('name','|S30'),('age','<i8'),('weight','<f8')])
>>> a = np.array([('Liming',24,63.9),('Mike',15,67.),('Jan',34,45.8)],dtype = persontype)
>>> a
array([('Liming', 24L, 63.9), ('Mike', 15L, 67.0), ('Jan', 34L, 45.8)], 
      dtype=[('name', 'S30'), ('age', '<i8'), ('weight', '<f8')])

|:忽略字节顺序class

<:低位字节在前

>:高位字节在前

结构体数组的取值方式和通常数组差很少,能够经过下标取得元素:

>>> a[0]
('Liming', 24L, 63.9)
>>> a[0:-1]
array([('Liming', 24L, 63.9), ('Mike', 15L, 67.0)], 
      dtype=[('name', 'S30'), ('age', '<i8'), ('weight', '<f8')])

咱们能够使用字段名做为下标获取对于的值

>>> a['name']
array(['Liming', 'Mike', 'Jan'], 
      dtype='|S30')
>>> a['age']
array([24, 15, 34], dtype=int64)
>>> a['weight']
array([ 63.9,  67. ,  45.8])

能够经过 a.tostring()或者a.tofile()方法能够将数组a以二进制方式写成字符串或者写入文件

相关文章
相关标签/搜索