遍历文本文件生成matplotlib热力图时报dtype错误如何解决？

问题原因 #

从文本文件读取的每行内容本质是字符串类型，即使内容和你手动写在代码里的嵌套列表完全一致，Python也不会自动将其解析为可用于数值计算的列表对象。matplotlib的imshow方法要求输入数值类型的二维数组，传入字符串就会触发dtype <U248 无法转换为float的报错。

修复方案 #

步骤1：导入安全的字面量解析模块 #

使用Python标准库的ast.literal_eval解析字符串格式的嵌套列表，它只会解析Python字面量（比如列表、字典、数值等），比直接用eval安全，不会执行恶意代码。

步骤2：修正读取行后的解析逻辑 #

把读取到的字符串行解析为嵌套列表，再转换为浮点型numpy数组传入imshow。

步骤3：修复文件名拼接的类型错误 #

原代码中suffix是整数，直接和字符串拼接会触发类型错误，需要转为字符串后再拼接。

步骤4：可选：循环中关闭画布释放内存 #

批量生成图片时如果不关闭画布，会持续占用内存，可能导致程序运行变慢甚至崩溃。

修正后完整代码 #

import ast
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

mpl.rc('image', cmap='inferno')
rows = ["Row 1", "Row 2", "Row 3", "Row 4"]
columns = ["Column 1", "Column 2", "Column 3", "Column 4", "Column 5"]
suffix = 10001
title = "Thermal Data From Coldplate in TVAC-3"

with open('tvac3_thermal_model.txt', 'r') as file:
    lines = file.readlines()
    lines = [line.rstrip() for line in lines]
    for line in lines:
        # 解析字符串为嵌套列表，再转成浮点型numpy数组
        heat_data = np.array(ast.literal_eval(line), dtype=np.float32)
        fig, ax = plt.subplots()
        im = ax.imshow(heat_data)
        ax.set_xticks(np.arange(len(columns)))
        ax.set_yticks(np.arange(len(rows)))
        ax.set_xticklabels(columns)
        ax.set_yticklabels(rows)
        ax.set_title(title)
        plt.setp(ax.get_xticklabels(), rotation=45, ha="right", rotation_mode="anchor")
        # 修正整数转字符串拼接的问题
        plt.savefig(f"frame{suffix}.png")
        # 关闭画布释放内存
        plt.close()
        suffix += 1

内容的提问来源于stack exchange，提问作者DBarnett