1. 首页
  2. IT资讯

Numpy 结构数组

“u003Cdivu003Eu003Cpu003E在C语言中我们可以通过struct关键字定义结构类型,结构中的字段占据连续的内存空间,每个结构体占用的内存大小都相同,因此可以很容易地定义结构数组。和C语言一样,在NumPy中也很容易对这种结构数组进行操作。u003Cu002Fpu003Eu003Cpu003E只要NumPy中的结构定义和C语言中的定义相同,NumPy就可以很方便地读取C语言的结构数组的二进制数据,转换为NumPy的结构数组。u003Cu002Fpu003Eu003Cdiv class=”pgc-img”u003Eu003Cimg src=”http:u002Fu002Fp1.pstatp.comu002Flargeu002Fdfic-imagehandleru002F54c8a2aa-088d-4290-9836-15319eefc7ad” img_width=”1200″ img_height=”800″ alt=”Numpy 结构数组” inline=”0″u003Eu003Cp class=”pgc-img-caption”u003Eu003Cu002Fpu003Eu003Cu002Fdivu003Eu003Cpu003E假设我们需要定义一个结构数组,它的每个元素都有name, age和weight字段。在NumPy中可以如下定义:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003Eimport numpy as npu003Cbru003Epersontype = np.dtype({‘names’:[‘name’, ‘age’, ‘weight’],’formats’:[‘S32′,’i’, ‘f’]})u003Cbru003Ea = np.array([(“Zhang”,32,75.5),(“Wang”,24,65.2)],dtype=persontype)u003Cbru003Eu003Cu002Fpreu003Eu003Cpu003E我们先创建一个dtype对象persontype,通过其字典参数描述结构类型的各个字段。字典有两个关键字:names,formats。每个关键字对应的值都是一个列表。names定义结构中的每个字段名,而formats则定义每个字段的类型:u003Cu002Fpu003Eu003Cpu003E• S32 : 32个字节的字符串类型,由于结构中的每个元素的大小必须固定,因此需要指定字符串的长度u003Cu002Fpu003Eu003Cpu003E• i : 32bit的整数类型,相当于np.int32u003Cu002Fpu003Eu003Cpu003E• f : 32bit的单精度浮点数类型,相当于np.float32u003Cu002Fpu003Eu003Cpu003E然后我们调用array函数创建数组,通过关键字参数dtype=persontype, 指定所创建的数组的元素类型为结构persontype。u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> a.dtypeu003Cbru003Edtype([(‘name’, ‘|S32’), (‘age’, ‘<i4’), (‘weight’, ‘<f4′)])u003Cbru003Eu003Cu002Fpreu003Eu003Cpu003E这里我们看到了另外一种描述结构类型的方法: 一个包含多个组元的列表,其中形如(字段名, 类型描述) 的组元描述了结构中的每个字段。类型描述前面为我们添加了`|’, `<‘ 等字符,这些字符用来描述字段值的字节顺序:u003Cu002Fpu003Eu003Cpu003E• | : 忽视字节顺序u003Cu002Fpu003Eu003Cpu003E• < : 低位字节在前u003Cu002Fpu003Eu003Cpu003E• > : 高位字节在前u003Cu002Fpu003Eu003Cpu003E结构数组的存取方式和一般数组相同,通过下标能够取得其中的元素,注意元素的值看上去像是组元,实际上它是一个结构:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> a[0]u003Cbru003E(‘Zhang’, 32, 75.5)u003Cbru003E>>> a[0].dtypeu003Cbru003Edtype([(‘name’, ‘|S32’), (‘age’, ‘<i4’), (‘weight’, ‘<f4’)])u003Cbru003Eu003Cu002Fpreu003Eu003Cpu003Ea[0]是一个结构元素,它和数组a共享内存数据,因此可以通过修改它的字段,改变原始数组中的对应字段:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> c = a[1]u003Cbru003E>>> c[“name”] = “Li”u003Cbru003E>>> a[1][“name”]u003Cbru003E”Li”u003Cbru003Eu003Cu002Fpreu003Eu003Cpu003E结构像字典一样可以通过字符串下标获取其对应的字段值:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> a[0][“name”]u003Cbru003E’Zhang’u003Cbru003Eu003Cu002Fpreu003Eu003Cpu003E我们不但可以获得结构元素的某个字段,还可以直接获得结构数组的字段,它返回的是原始数组的视图,因此可以通过修改b[0]改变a[0][”age”]:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> b=a[:][“age”] # 或者a[“age”]u003Cbru003E>>> bu003Cbru003Earray([32, 24])u003Cbru003E>>> b[0] = 40u003Cbru003E>>> a[0][“age”]u003Cbru003E40u003Cbru003Eu003Cu002Fpreu003Eu003Cpu003E通过调用a.tostring或者a.tofile方法,可以直接输出数组a的二进制形式:u003Cu002Fpu003Eu003Cpu003E>>> a.tofile(“test.bin”)u003Cu002Fpu003Eu003Cpu003E利用下面的C语言程序可以将test.bin文件中的数据读取出来。u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E#include <stdio.h>u003Cbru003Estruct personu003Cbru003E{u003Cbru003E char name[32];u003Cbru003E int age;u003Cbru003E float weight;u003Cbru003E};u003Cbru003Estruct person p[2];u003Cbru003Evoid main ()u003Cbru003E{u003Cbru003E FILE *fp;u003Cbru003E int i;u003Cbru003E fp=fopen(“test.bin”,”rb”);u003Cbru003E fread(p, sizeof(struct person), 2, fp);u003Cbru003E fclose(fp);u003Cbru003E for(i=0;i<2;i++)u003Cbru003E printf(“%s %d %f\n”, p[i].name, p[i].age, p[i].weight);u003Cbru003E getchar();u003Cbru003E}u003Cbru003Eu003Cu002Fpreu003Eu003Cp class=”ql-align-justify”u003E结构类型中可以包括其它的结构类型,下面的语句创建一个有一个字段f1的结构,f1的值是另外一个结构,它有字段f2,其类型为16bit整数。u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> np.dtype([(‘f1’, [(‘f2’, np.int16)])])u003Cbru003Edtype([(‘f1’, [(‘f2’, ‘<i2’)])])u003Cbru003Eu003Cu002Fpreu003Eu003Cp class=”ql-align-justify”u003E当某个字段类型为数组时,用组元的第三个参数表示,下面描述的f1字段是一个shape为(2,3)的双精度浮点数组:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> np.dtype([(‘f0’, ‘i4’), (‘f1’, ‘f8’, (2, 3))])u003Cbru003Edtype([(‘f0’, ‘<i4’), (‘f1’, ‘<f8’, (2, 3))])u003Cbru003Eu003Cu002Fpreu003Eu003Cp class=”ql-align-justify”u003E用下面的字典参数也可以定义结构类型,字典的关键字为结构中字段名,值为字段的类型描述,但是由于字典的关键字是没有顺序的,因此字段的顺序需要在类型描述中给出,类型描述是一个组元,它的第二个值给出字段的字节为单位的偏移量,例如age字段的偏移量为25个字节:u003Cu002Fpu003Eu003Cpre class=”ql-align-justify”u003E>>> np.dtype({‘surname’:(‘S25′,0),’age’:(np.uint8,25)})u003Cbru003Edtype([(‘surname’, ‘|S25’), (‘age’, ‘|u1’)])u003Cbru003Eu003Cu002Fpreu003Eu003Cp class=”ql-align-justify”u003E关于内存对齐:u003Cu002Fpu003Eu003Cp class=”ql-align-justify”u003EC语言的结构体为了内存寻址方便,会自动的添加一些填充用的字节,这叫做内存对齐。例如如果把下面的name[32]改为name[30]的话,由于内存对齐问题,在name和age中间会填补两个字节,最终的结构体大小不会改变。因此如果numpy中的所配置的内存大小不符合C语言的对齐规范的话,将会出现数据错位。为了解决这个问题,在创建dtype对象时,可以传递参数align=True,这样numpy的结构数组的内存对齐和C语言的结构体就一致了。u003Cu002Fpu003Eu003Cu002Fdivu003E”

原文始发于:Numpy 结构数组

主题测试文章,只做测试使用。发布者:玩家L-,转转请注明出处:http://www.cxybcw.com/10610.html

联系我们

13687733322

在线咨询:点击这里给我发消息

邮件:1877088071@qq.com

工作时间:周一至周五,9:30-18:30,节假日休息

QR code