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

Question

尝试在 python 中读取二进制文件。从数据集页面：

像素存储为无符号字符（1 个字节），取值范围为 0 到 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 字节的不可变数组（str在 Python 2 中称为 a ）。如果您需要它是可变的，您可以使用array模块及其能够存储各种原语的数组类型，包括无符号字节（由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)]