如何解决Numba运行时异常：Cannot determine Numba type of <class 'type'>

嘿，我来帮你捋清楚问题根源，以及怎么搞定这个异常。

你遇到的这个报错，核心原因是Numba的@njit纯编译模式对dtype=object的numpy数组支持极差。你要存储的是不同形状的子数组，而object数组里的元素类型、形状都是动态变化的——Numba需要在编译阶段就确定所有变量的类型，这种动态不确定的情况它根本没法推断，所以直接抛出了类型错误。

至于你尝试换成dtype=float64也不行，是因为每个元素是二维数组，不是单个浮点数，类型完全不匹配，自然也会报错。

解决思路：用列表替代object数组收集结果 #

Numba对Python列表的动态存储支持要好得多，哪怕列表里的元素是不同形状的数值数组，只要它们的基础数值类型一致（比如都是float64），Numba就能正常处理。

下面是修改后的可运行代码：

import numpy as np
from numba import njit
from timeit import timeit

@njit
def required_numba_function():
    nRows = 151
    nCols = 151
    nFrames = 24
    total_elements = nRows * nCols * nFrames
    # 用列表替代object数组，Numba能很好处理这种动态存储
    result_list = [None] * total_elements
    
    for frame in range(nFrames):
        for row in range(nRows):
            for col in range(nCols):
                size_rows = np.random.randint(8, 15)
                size_cols = np.random.randint(2, 6)
                args = np.random.normal(3, 2.5, size=(size_rows, size_cols))
                flat_idx = frame * (nRows * nCols) + (row * nCols + col)
                result_list[flat_idx] = args
    
    # 如果需要numpy数组格式，最后转成object dtype数组即可
    return np.array(result_list, dtype=object)

if __name__ == "__main__":
    result = required_numba_function()
    print(f"结果总长度: {len(result)}")
    print(f"第一个子数组形状: {result[0].shape}")

为什么这样能行？ #

1. 列表result_list是动态结构，Numba在编译时不需要提前知道每个元素的具体形状，只要知道元素是float64类型的数组就行，这完全符合它的类型推断要求。
2. 最后转成object dtype数组的操作，要么在函数内部完成（Numba会把这部分当成运行时操作），要么在函数外部做都可以，不会触发类型推断问题。

额外提醒 #

如果后续你需要对这些子数组做批量运算，因为形状不固定，可能得逐个处理——毕竟这种场景下，没有完美的“批量结构”能适配可变形状的子数组，列表/object数组是目前最贴合你需求的方案。

备注：内容来源于stack exchange，提问作者skm