使用Python编写的load_npz()函数

以下是一个使用Python编写的load_npz()函数，并附带一个使用例子:

import numpy as np
import scipy.sparse as sp

def load_npz(file_path):
    """
    加载以npz格式保存的稀疏矩阵数据文件

    参数：
    file_path (str): 文件的路径

    返回：
    data (scipy.sparse.csr_matrix): 加载的稀疏矩阵数据
    """
    data = np.load(file_path)
    if isinstance(data, np.lib.npyio.NpzFile):
        data = data[list(data.files)[0]]
    if isinstance(data, np.ndarray):
        data = sp.csr_matrix(data)
    return data

# 使用示例
file_path = 'data_sparse_matrix.npz'
sparse_matrix = load_npz(file_path)
print(sparse_matrix)

在上面的示例中，load_npz()函数用于加载以npz格式保存的稀疏矩阵数据文件。该函数接收一个参数file_path，表示文件的路径。函数首先使用numpy的np.load()函数加载文件，然后检查加载的数据类型是否为np.lib.npyio.NpzFile类型，如果是，则获取该类型中的 个数据项。接着，函数判断数据类型是否为numpy的ndarray类型，如果是，则将其转换为scipy的csr_matrix类型。返回加载的稀疏矩阵数据。

在使用例子中，我们定义了一个文件路径file_path，然后调用load_npz()函数加载该文件，并将加载的稀疏矩阵数据赋值给变量sparse_matrix。最后，输出加载的稀疏矩阵数据。

希望以上解释对您有帮助。