# 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了)