将一个32×32的二进制图像矩阵转换为1×1024的向量。先创建一个1×1024的NumPy数组,然后循环读出文件的前32行,并将 每行的前32个字符值存储在数组中。代码为:
from numpy import *
def img2vector (filename):
returnVect = zeros((1, 1024)) #创建一个1*1024的数组
fr = open(filename)
for i in range(32):
lineStr = fr.readline() #每次读入一行
for j in range(32):
returnVect[0, 32*i+j] = int(lineStr[j])
return returnVect