如何使用Rust加速Python程序

如何使用Rust加速Python程序(2)

时间：2024-02-02 21:00:17 来源：IT袋作者：马勇

导读：如何使用Rust加速Python程序，是一个包清单。它以TOML格式编写，包含编译包所需的元数据。在示例中，只需要将相对名称更改为 rustFib ，并保留其他设置的默认值。 [package]name = "rus

如何使用Rust加速Python程序

是一个包清单。它以TOML格式编写，包含编译包所需的元数据。

在示例中，只需要将相对名称更改为rustFib，并保留其他设置的默认值。

[package]
name = "rustFib"
version = "0.1.0"
edition = "2021"
[lib]
name = "rustFib"
crate-type = ["cdylib"]
[dependencies]
pyo3 = "0.19.0"

可以在lib.rs文件中编写Rust代码，并将其作为Python函数的替代品来提高程序的性能。：

use pyo3::prelude::*;
/// Rust中实现的Python函数
#[pyfunction]
fn fib(n: i32) -> u64 {
    match n {
        1 | 2 => 1,
        _ => fib(n - 1) + fib(n - 2)
    }
}
/// 将其作为Python模块。
#[pymodule]
fn rustFib(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_function(wrap_pyfunction!(fib, m)?)?;
    Ok(())
}

最后，只需要执行命令来编译Rust代码：

maturin develop

已经成功地用Rust构建了名为rustFib的Python包。它的速度会与原始Rust程序一样快吗？

现在在Python中使用看看：

import time
from rustFib import fib
def main(test_times=50):
    start = time.time()
    for _ in range(test_times):
        fib(30)
    print(f"Total time spent: {time.time() - start} s")
main()
# Total time spent: 0.17684102058410645 s

如上面的代码所示，总共花费的时间现在是0.176 ms。真的使Python程序像Rust一样快啦！

以上IT袋网网介绍的如何使用Rust加速Python程序的IT小经验，希望能为您在生活中带来帮助！