详解Python中的array数组模块相关使用

初始化
array实例化可以提供一个参数来描述允许那种数据类型，还可以有一个初始的数据序列存储在数组中。

import array
import binascii
s = 'This is the array.'
a = array.array('c',s)
print 'As string:',s
print 'As array :',a
print 'As hex  :',binascii.hexlify(a)

数组配置为包含一个字节序列，用一个简单的字符串初始化。

>>> ================================ RESTART ================================
>>> 
As string: This is the array.
As array : array('c','This is the array.')
As hex  : 54686973206973207468652061727261792e

处理数组
类似于其他python序列，可以采用同样方式扩展和处理array。

import array
import pprint
a = array.array('i',xrange(3))
print 'Initial :',a
a.extend(xrange(3))
print 'Extended:',a
print 'slice: :',a[2:5]
print 'Itetator:'
print list(enumerate(a))

支持的操作包括分片，迭代以及向末尾增加元素。

>>> ================================ RESTART ================================
>>> 
Initial : array('i',[0,1,2])
Extended: array('i',2,2])
slice: : array('i',[2,1])
Itetator:
[(0,0),(1,1),(2,2),(3,(4,(5,2)]

数组和文件
可以使用高效读/写文件的专用内置方法将数组的内容写入文件或从文件读取数组。

import array
import binascii
import tempfile

a = array.array('i',xrange(5))
print 'A1: ',a
output = tempfile.NamedTemporaryFile()
a.tofile(output.file)
output.flush

with open(output.name,'rb') as input:
  raw_input = input.read()
  print 'Raw Contents:',binascii.hexlify(raw_data)

  input.seek(0)
  a2 = array.array('i')
  a2.fromfile(input,len(a))
  print 'A2: ',a2

   

候选字节顺序
如果数组中的数据没有采用固有的字节顺序，或者在发送到一个采用不同字节顺序的系统前需要交换顺序，可以在python转换整个数组而无须迭代处理每个元素。

import array
import binascii

def to_hex(a):
  chars_per_item = a.itemsize * 2
  hex_version = binascii.hexlify(a)
  num_chunks = len(hex_version) / chars_per_item
  for i in xrange(num_chunks):
    start = i * chars_per_item
    end = start + chars_per_item
    yield hex_version[start:end]

a1 = array.array('i',xrange(5))
a2 = array.array('i',xrange(5))
a2.byteswap()

fmt = '%10s %10s %10s %10s'
print fmt % ('A1_hex','A1','A2_hex','A2')
print fmt % (('-' * 10,) * 4)
for value in zip(to_hex(a1),a1,to_hex(a2),a2):
  print fmt % value

byteswap()会交换C数组中元素的字节顺序，比在python中循环处理数据高效的多。   

>>> ================================ RESTART ================================
>>> 
  A1_hex     A1   A2_hex     A2
---------- ---------- ---------- ----------
 00000000     0  00000000     0
 01000000     1  00000001  16777216
 02000000     2  00000002  33554432
 03000000     3  00000003  50331648
 04000000     4  00000004  67108864