【一分钟】熟悉np.load()和np.save()的使用

引言

NumPy是数据科学中非常重要的Python库。NumPy提供了高性能的多维数组对象和用于处理这些数组的工具。在NumPy中，我们可以使用np.save()将数组保存到文件中，也可以使用np.load()从文件中读取数组数据。这些函数非常有利于我们在不同的Python程序之间共享数据或保存数据以供未来使用。

保存NumPy数组到文件

我们可以使用np.save()函数将NumPy数组保存到文件中。该函数需要两个参数：文件名和要保存的数组。以下是一个示例代码，演示如何将NumPy数组保存到文件中：

import numpy as np

# 创建一个NumPy数组
arr = np.array([1, 2, 3, 4, 5])

# 将数组保存到文件中
np.save('my_array.npy', arr)

从文件中加载NumPy数组

我们可以使用np.load()函数从文件中加载NumPy数组。该函数需要一个参数：要加载的文件名。

以下是一个示例代码，演示如何从文件中加载NumPy数组：

import numpy as np

# 从文件中加载数组
arr = np.load('my_array.npy')

# 打印数组
print(arr)

小结

在本文中，小哥向你介绍了如何使用NumPy库中的np.save()和np.load()函数将NumPy数组保存到文件中并从文件中加载数组。这对于在不同的Python程序之间共享数据或在不同的时间点保存数据以供以后使用非常有用。