Numba parallel模式下如何不预分配定义数组及索引赋值问题

按需分配数组报错问题 #

你遇到的变量类型错误确实和parallel=True参数直接相关。Numba的并行编译模式要求所有变量在编译阶段就确定固定类型，不支持「要么为None要么为数组」的动态类型变化。
有两种可行的解决方案：

• 不需要并行加速时直接关闭parallel参数：非并行模式下Numba支持动态类型推断，可以正常用None初始化变量。
• 需要保留并行能力时，用空数组替代None作为初始值：用ary = np.empty((0, 2), dtype=np.float64)初始化，此时变量类型固定为二维浮点数组，不需要使用时直接返回空数组即可，几乎不占用内存，也不会触发类型错误。

切片赋值只更新第一行问题 #

这个是Numba 0.56.x及更早版本的已知bug，并行模式下结合切片赋值时会出现索引计算错误，导致所有写入操作都落到第一行。ary[i]属于按行索引赋值，没有触发该切片bug，所以运行正常。
你可以选择两种处理方式：

• 升级Numba到0.57.0及以上版本，该bug已经在后续版本中修复。
• 保持当前版本不变，统一使用ary[i]的按行赋值写法即可。

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