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,中间以空格隔开,数据图片显示如下:
Python编程实现,将矩阵数据转换成为一个图像_第1张图片
运行代码处理后的结果如下图:
Python编程实现,将矩阵数据转换成为一个图像_第2张图片
(如果你了解生信领域的话,就知道结果图为一个contactMap了)

你可能感兴趣的:(Python学习)