numpy矩阵存储lambda函数/数学表达式报错及字符串截断问题求助

问题原因说明 #

1. Numpy常规数值类型数组不支持存储Python可调用对象（如lambda函数），所以直接存入lambda会触发类型错误。
2. np.chararray默认会设置固定的字符串长度，未显式指定itemsize参数时默认长度仅为1，长于1位的表达式会被直接截断，因此x**2最终仅存入了x。

解决方法 #

根据你的后续使用需求，可以选择以下两种方案：

方案1：存储lambda函数，方便后续直接传值计算 #

创建数组时指定dtype=object，即可支持存储任意Python对象，包括lambda函数：

import numpy as np

n, m = 2, 2 # 替换为你的实际矩阵维度
# 初始化object类型的空数组
matrix_A = np.empty((n, m), dtype=object)

# 从tkinter控件读取内容并转成lambda函数存入
for i in range(n):
    for j in range(m):
        expr_str = text_var_A[i][j].get()
        # 将表达式字符串转为可调用的lambda函数
        matrix_A[i][j] = eval(f"lambda x: {expr_str}")

# 后续传值计算示例，比如x=3时计算整个矩阵的数值
x_val = 3
calc_result = np.vectorize(lambda f: f(x_val))(matrix_A)
# 输出结果为[[5 9],[1 3]]，符合预期结构

注意：如果表达式来自用户输入，使用eval存在代码注入风险，生产环境需额外做表达式安全校验。

方案2：仅存储字符串表达式 #

放弃使用np.chararray，改用指定长度的Unicode字符串类型或object类型数组，即可避免字符串截断问题：

import numpy as np

n, m = 2, 2
# 方法1：指定最大字符串长度，按需调整长度数值（示例为支持最长30个字符的表达式）
matrix_A = np.empty((n, m), dtype='U30')
# 方法2：用object类型，完全不限制字符串长度
# matrix_A = np.empty((n, m), dtype=object)

for i in range(n):
    for j in range(m):
        current_f = text_var_A[i][j].get()
        matrix_A[i][j] = current_f

内容的提问来源于stack exchange，提问作者Максим Левченко