如何将二进制文件数据读入数组？

Question

尝试读取 python 中的二进制文件。来自 dataset page:

The pixels are stored as unsigned chars (1 byte) and take values from 0 to 255

我试过以下方法，它打印 (0,)，而不是 784,000 位数字的数组。

# -*- coding: utf8 -*-
# Processed MNIST dataset (http://cis.jhu.edu/~sachin/digit/digit.html)
import struct

f = open('data/data0', mode='rb')
data = []

print struct.unpack('<i', f.read(4))

如何将此二进制文件读入 784,000 位数组（28 字节 x 28 字节 x 1k 样本）或 28x28x1000 3D 数组。我以前从未使用过二进制文件，所以很困惑！

Answer 1

f.read() 将为您提供一个 784,000 字节的不可变数组（在 Python 2 中称为 str）。如果你需要它是可变的，你可以使用 array module 及其能够存储各种基元的数组类型，无符号字节（由 B 代码表示）包括：

from array import array

data = array('B')

with open('data/data0', 'rb') as f:
    data.fromfile(f, 784000)

这可以根据需要切片：

EXAMPLE_SIZE = 24 * 24
examples = [data[s:s + EXAMPLE_SIZE] for s in xrange(0, len(a), EXAMPLE_SIZE)]

如何将二进制文件数据读入数组？

How to read binary file data into arrays?

python

arrays

file

binaries

python-2.7