博客

Python 存取npy格式数据实例

  • Post category:Python

Python存取.npy格式数据实例

.npy是numpy库中所使用的一种二进制数据格式，用于存储多维数组（ndarrays）。

在Python中，通过使用Numpy中的load和save函数可以非常方便地存取.npy格式数据，这些函数的调用方式非常简单。

1. 存储npy数据

我们使用numpy库生成一个简单的数组：

import numpy as np

data = np.array([[1, 2], [3, 4]])

将data数组存储为“data.npy”文件：

np.save("data.npy", data)

这样就可以将数组以npy格式存储在当前工作目录中的”data.npy”文件中。如果想要将数组存储在其他目录下，只需将文件路径传递给save函数即可。

2. 读取npy数据

使用load函数可以将.npy文件读入到Python中：

loaded_data = np.load("data.npy")

这样就可以将”data.npy”文件中的数据读入到Python中并存储在loaded_data变量中。可以通过打印loaded_data来查看输出结果：

print(loaded_data)

输出结果为：

[[1 2]
 [3 4]]

3. 示例1 – 存储和读取文本数据

import numpy as np

# 定义数据
data = np.array([[1, 2], [3, 4]])

# 存储数据
np.save("data.npy", data)

# 读取数据
loaded_data = np.load("data.npy")

# 打印结果
print(loaded_data)

这段代码会在控制台输出以下内容：

[[1 2]
 [3 4]]

4. 示例2 – 存储和读取图像数据

import numpy as np
from PIL import Image

# 加载图像文件
img = Image.open("test.png")

# 将图像转化为numpy数组
data = np.array(img)

# 存储数据
np.save("img.npy", data)

# 读取数据
loaded_data = np.load("img.npy")

# 将npy数组转化为图像
loaded_img = Image.fromarray(loaded_data)

# 显示图像
loaded_img.show()

这段代码会将“test.png”文件转化为二维numpy数组，并将其存储在“img.npy”文件中。然后，通过load函数加载“img.npy”文件，并将其转化为图像，并将图像显示在屏幕上。

注意：在使用PIL库时，需要安装pillow库。可以通过以下命令安装pillow库：

pip install pillow

结论

在Numpy中存取.npy格式数据非常容易，只需要使用save和load函数即可。同时，也可以将不同类型的数据存储为.npy文件。

Tags: ,