Python中使用np.load

在Python中，可以使用numpy库中的load()函数来读取存储在.npy文件中的numpy数组。

基本语法

import numpy as np

arr = np.load('file.npy')

参数说明

• file: str or file-like object
  文件名或类似文件对象的对象

示例

import numpy as np

# 生成一个numpy数组
arr = np.arange(9).reshape(3,3)

# 将数组存储为.npy文件
np.save('arr.npy', arr)

# 使用load()函数读取.npy文件
arr_load = np.load('arr.npy')

# 输出读取的数组
print(arr_load)

输出结果为：

[[0 1 2]
 [3 4 5]
 [6 7 8]]

注意： 与np.load()对应的还有np.save()函数，用于将numpy数组保存为.npy文件。