在实际应用中，我们有时候会遇到“Python读取npy文件怎样实现，代码是什么”这样的问题，我们该怎样来处理呢？下文给大家介绍了解决方法，希望这篇“Python读取npy文件怎样实现，代码是什么”文章能帮助大家解决问题。

Numpy binary files (NPY, NPZ)

注：.npy文件是numpy专用的二进制文件。

1. 读取与保存

import numpy as np

arr = np.array([[1, 2, 3],
               [4, 5, 6]])
np.save('weight.npy', arr)

loadData = np.load('weight.npy')

print("----type----")
print(type(loadData))
print("----shape----")
print(loadData.shape)
print("----data----")
print(loadData)

至于具体API参见：https://docs.scipy.org/doc/numpy/index.html

2. 实战案例

在深度神经网络训练过程中通常需要读取预训练权重，预训练权重通常是 .npy文件，比如vgg16.npy（https://pan.baidu.com/s/1Ru5FJVSPjYTHZwlmzRwRvQ   提取码：ygxw）。本次就以分析vgg16.npy为例进行说明。

import numpy as np

# 注意编码方式
pre_train = np.load("vgg16.npy", allow_pickle=True, encoding="latin1")

print("------type-------")
print(type(pre_train))
print("------shape-------")
print(pre_train.shape)
print("------data-------")
print(pre_train)

这是个啥？为啥shape没有？ 但是可以看出来 pre_train 里元素应该是一个字典，我们尝试取出来。

注：ndarray.item()是复制数组中的一个元素，并将其返回。具体语法参见：https://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.item.html?highlight=item#numpy.ndarray.item

import numpy as np

pre_train = np.load("vgg16.npy", allow_pickle=True, encoding="latin1")

data_dic = pre_train.item()

print("------type-------")
print(type(data_dic))
print("------conv1_1  data-------")
print(data_dic['conv1_1'])   # 返回一个列表，该列表有两个array，表示conv1_1的权重w与偏置b
print("------conv1_1  shape-------")
print((data_dic['conv1_1'][0]).shape)

看看结果：

可以发现，这是第一个卷积层的权重参数，输入channel是3，输出channel是64。

附：python中 .npy文件的读写操作实例

numpy中的二进制文件的读写:

save

np.save ("./文件名", 数组名)：以二进制的格式保存数据

load

np.load("./文件名.npy")： 函数是从二进制的文件中读取数据

savez

np.savez(’./文件名’,数组名1,数组名2,…)：savez 函数可以将多个数组保存到一个文件中

(1)save操作

import numpy as np
a=np.arange(5)
np.save('get.npy',a)

(2)load操作

import numpy as np
a=np.load('load.npy')
print(a)

(3)savez操作

import numpy as np
a=np.arange(3)
b=np.arange(4)
c=np.arange(5)
np.savez('array_save.npz',a,b,c)  多个ndarray类型的数组

总结

到此，关于“Python读取npy文件怎样实现，代码是什么”的学习就结束了，希望能够解决大家的疑惑，另外大家动手实践也很重要，对大家加深理解和学习很有帮助。如果想要学习更多的相关知识，欢迎关注群英网络，小编每天都会给大家分享实用的文章!

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：mmqy2019@163.com进行举报，并提供相关证据，查实之后，将立刻删除涉嫌侵权内容。

猜你喜欢

• 基于Python如何写通讯录管理系统，并实现基础的操作

  这篇文章主要为大家详细介绍了python实现简单通讯录管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

• 用Python怎样写仓库管理系统，代码是什么

  本文实例为大家分享了Python实现仓库管理系统的具体代码，供大家参考，具体内容如下注意：在Linux环境运行代码#!/usr/bin/env python# -*- coding:utf-8 -*-# @FileName  :store

• Python append方法如何使用，什么坑要注意

  这篇文章主要介绍了Python之列表的append()方法最容易踩的坑及解决，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

• Python中List遍历怎样做，如何遍历整个列表的

  这篇文章主要介绍了Python中列表(List) 的三种遍历(序号和值)方法小结，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

• 用python的seed()方法怎样生成随机数

  如果要考察某公司的牛奶产品质量，可以从100袋牛奶中抽取30袋，在随机数表中选中一数，并用向上、下、左、右不同的读法组成30个数，并按牛奶的标号进行检测，虽然麻烦，但很常用。在日常生活中，随机数起着很大的作用，所以很多人会专门去寻找随机数生成器。