Python中对文件夹下的特定格式图像全部读取并转化为数组保存（也可转化为txt文件）

 D:\test 的目录

2016/11/03  21:19             <DIR>          .
2016/11/03  21:19             <DIR>          ..
2016/11/03  21:22             1,596      num7.txt
2016/11/03  21:22             1,010      test.py
2016/07/02  20:16             1,862      TestImage_1.bmp
2016/07/02  20:16             1,862      TestImage_2.bmp
2016/07/02  20:16             1,862      TestImage_2222.jpg
               5 个文件          8,192 字节
               2 个目录 444,028,071,936 可用字节

#-*- coding: utf-8 -*- 
import os
import numpy
from PIL import Image   #导入Image模块
from pylab import *     #导入savetxt模块


#以下代码看可以读取文件夹下所有文件
# def getAllImages(folder):
#     assert os.path.exists(folder)
#     assert os.path.isdir(folder)
#     imageList = os.listdir(folder)
#     imageList = [os.path.abspath(item) for item in imageList if os.path.isfile(os.path.join(folder, item))]
#     return imageList

# print getAllImages(r"D:\\test")



def get_imlist(path):   #此函数读取特定文件夹下的bmp格式图像

    return [os.path.join(path,f) for f in os.listdir(path) if f.endswith('.bmp')]

c=get_imlist(r"D:\test")    #r""是防止字符串转译
print c     #这里以list形式输出bmp格式的所有图像（带路径）
d=len(c)    #这可以以输出图像个数


data=numpy.empty((d,28*28)) #建立d*（28*28）的矩阵
while d>0:
    img=Image.open(c[d-1])  #打开图像
    #img_ndarray=numpy.asarray(img)
    img_ndarray=numpy.asarray(img,dtype='float64')/256  #将图像转化为数组并将像素转化到0-1之间
    data[d-1]=numpy.ndarray.flatten(img_ndarray)    #将图像的矩阵形式转化为一维数组保存到data中
    d=d-1
print data

A=numpy.array(data[0]).reshape(28,28)   #将一维数组转化为矩28*28矩阵
#print A
savetxt('num7.txt',A,fmt="%.0f") #将矩阵保存到txt文件中