Mojo中使用Python.import_module导入numpy包报错求助

错误原因分析 #

1. 静态类型约束：Mojo的fn是静态类型函数，所有变量必须显式声明类型。你代码中的np未指定类型，导致编译器无法识别，触发use of unknown declaration 'np'错误。
2. Python对象类型匹配：通过Python.import_module导入的Python模块属于PythonObject类型，必须明确标注该类型才能在fn中正常使用。

修正后的代码 #

from python import Python
from python import PythonObject

fn use_array() raises:
    # 显式声明np的类型为PythonObject
    let np: PythonObject = Python.import_module("numpy")
    
    # numpy返回的数组也是PythonObject类型，同样需要显式标注
    let array: PythonObject = np.array([1, 2, 3])
    print(array)

use_array()

关键修改点说明 #

• 导入PythonObject类型：需从python模块中导入该类型，用于标注Python模块和对象的类型。
• 显式变量类型声明：在let np和let array后添加: PythonObject，满足Mojo静态类型函数的要求。
• 保留raises关键字：导入Python模块可能抛出异常，fn必须声明raises来处理潜在错误。

额外注意事项 #

• 缩进规范：Mojo对缩进要求与Python一致，确保代码块缩进统一（建议用4个空格），错误缩进也可能触发解析错误。
• 动态函数def的用法：如果用def定义动态函数，可省略类型声明，但性能不如fn，示例如下：

from python import Python

def use_array():
    let np = Python.import_module("numpy")
    let array = np.array([1, 2, 3])
    print(array)

use_array()

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