17、Python模块与Rust结合：NumPy的高效应用

Python模块与Rust结合：NumPy的高效应用

1. 纯Python向量相加

在向量相加操作中，我们可以使用纯Python代码来实现，并对其进行计时。以下是具体的代码示例：

import time

def python_function(total_vector_size: int) -> float:
    t1 = time.time()
    first_vector = range(total_vector_size)
    second_vector = range(total_vector_size)
    sum_vector = [first_vector[i] + second_vector[i] for i in range(len(second_vector))]
    return time.time() - t1

通过运行上述代码，我们可以得到纯Python实现向量相加所需的时间。为了对比NumPy和纯Python的性能，我们可以同时运行两者的函数并绘制图表：

import matplotlib.pyplot as plt
import numpy as np

def numpy_function(total_vector_size: int) -> float:
    t1 = time.time()
    first_vector = np.arange(total_vector_size)
    second_vector = np.arange(t