如何将TXT中的数组数据转换为灰度图像？Python实现求助

原代码的核心问题 #

• 数组赋值错误：A[A_row:] = list[0:columns] 未指定单行赋值，应该写成 A[A_row, :]，否则会把整行数据覆盖从A_row开始的所有行。
• Image.fromarray 参数错误：传入了numpy的array模块名，而非读取到的数组变量A。
• 数据类型不匹配：直接用int类型数组转图像可能显示异常，灰度图需要0-255范围的uint8类型数据。
• 文件读取未规范处理：未手动关闭文件，推荐用with语句自动管理文件句柄。

修正后的代码 #

import numpy as np
from PIL import Image

def read_txt_to_array(txtname):
    # 用with语句自动关闭文件，避免资源泄漏
    with open(txtname, 'r') as f:
        lines = f.readlines()
    
    rows = len(lines)
    # 取第一行获取列数（假设所有行列数与你描述的5362一致）
    cols = len(lines[0].strip('\n').split(' '))
    
    # 初始化float类型数组，方便后续归一化处理
    data_array = np.zeros((rows, cols), dtype=np.float32)
    
    for row_idx, line in enumerate(lines):
        # 分割每行数据并转为数值类型
        row_data = list(map(float, line.strip('\n').split(' ')))
        data_array[row_idx, :] = row_data
    
    # 将原始数据归一化到0-255范围，转为灰度图要求的uint8类型
    data_array = (data_array - data_array.min()) / (data_array.max() - data_array.min()) * 255
    data_array = data_array.astype(np.uint8)
    
    return data_array

# 读取文本数据
img_data = read_txt_to_array('oop.txt')
# 直接生成灰度图像（mode='L'指定为灰度模式）
img = Image.fromarray(img_data, mode='L')
# 保存图像
img.save('T7.png')

关键说明 #

• 归一化处理：如果你的原始数据值不在0-255范围内，必须做归一化，否则PIL无法正确渲染灰度图。若数据本身已是0-255的整数，可跳过归一化步骤，直接将数组转为uint8类型。
• 行列数校验：代码默认所有行列数一致，若存在行数据列数不符的情况，需要提前清理文本数据。
• 模块导入规范：用import numpy as np替代from numpy import *，避免命名冲突，也是Python的标准写法。

内容的提问来源于stack exchange，提问作者純真丁一郎