Python编程实现，将矩阵数据转换成为一个图像

# coding=gbk
# 实现读取一个TXT文件，将文件中的数据存放在一个列表中，
# 再将列表逐渐转换为数组和矩阵
# 最后利用矩阵中的数据，将其以图像的形式呈现出来

from PIL import Image
import numpy as np

# 读取本地文件，文件格式为txt,将文件中的数据转存在一个list列表中
def readfile(filename):
    with open(filename, 'r') as f:
        list1 = []
        for line in f.readlines():
            line_str = line.strip()
            for element in line_str:
                if element != " ":
                    list1.append(int(element))
    return list1


if __name__ == '__main__':
    list_result = readfile("target_matrix.txt")
    # 测试的txt中，只有0和1，目标是把1显示为“黑色”，0显示为“白色”；
    # 所以将列表中的1替换为0，而0替换为255
    for i in range(0, len(list_result)):
        if list_result[i] == 1:
            list_result[i] = 0
        else:
            list_result[i] = 255
    # 再利用numpy将列表包装为数组
    array1 = np.array(list_result)
    # 进一步将array包装成矩阵
    data = np.matrix(array1)
    # 重新reshape一个矩阵为一个方阵
    data = np.reshape(data, (280, 280))
    # 调用Image的formarray方法将矩阵数据转换为图像PIL类型的数据
    new_map = Image.fromarray(data)
    # 显示图像
    new_map.show()

说明：读取的文件为TXT类型的文件，文件中只包含0和1，中间以空格隔开，数据图片显示如下：

运行代码处理后的结果如下图：

（如果你了解生信领域的话，就知道结果图为一个contactMap了）