Python数组的使用方法

liftword5个月前 (12-06)技术文章55

当想要存储单一类型的数据时,数组是比列表更好的选择。因为数组底层存储的是二进制数据,而列表存储的是对象。

有一个简单的对比:

数组从二进制文件中读取千万级数据,只需要0.1秒,比文本文件快60倍,写入快7倍,占用的存储空间减少60%。

1.创建数组

使用数组首先要导入array模块,使用array.array创建数组。

第一个参数是指定数据类型,'d'代表是double型数据。

第二个参数是指定数组元素。

Bash
import array
my_array = array.array('d', (i for i in range(1000)))
print(my_array)
print(my_array[100])

2.将数组数据存储到二进制文件中

使用tofile函数将数据存储到二进制文件中。

Bash
import array
my_array = array.array('d', (i for i in range(1000)))

file = open('array.bin', 'wb')
my_array.tofile(file)

3.从二进制文件中读取数组数据

使用fromfile函数从二进制文件中读取数组数据,第一个参数是文件对象,第二个参数是要读取的数量。

Bash
import array
my_array_i = array.array('d')
file = open('array.bin', 'rb')
my_array_i.fromfile(file,1000)
print(my_array_i)
print(my_array_i[100])