用python读取npy文件
时间: 2023-06-03 20:02:54 浏览: 503
可以使用 numpy.load() 函数来读取 .npy 文件,示例代码如下:
import numpy as np
# 读取 .npy 文件
data = np.load('data.npy')
# 打印数据
print(data)
相关问题
python读取npy多个文件
### 如何使用 Python 和 NumPy 读取多个 `.npy` 文件
为了实现批量处理多个 `.npy` 文件的功能,可以利用 `numpy.load()` 方法逐个加载这些文件。以下是具体方法:
#### 使用循环遍历目录中的所有 `.npy` 文件
可以通过标准库模块 `os` 或更高级的工具如 `glob` 来获取指定路径下的所有 `.npy` 文件名列表[^1]。
```python
import os
import numpy as np
# 定义存储 .npy 文件的目录
directory = './path/to/npy/files'
# 获取该目录下所有的 .npy 文件
npy_files = [f for f in os.listdir(directory) if f.endswith('.npy')]
data_list = []
for file_name in npy_files:
full_path = os.path.join(directory, file_name)
data = np.load(full_path) # 加载单个.npy文件的内容
data_list.append(data)
print(f"成功加载 {len(data_list)} 个 .npy 文件")
```
上述代码通过 `os.listdir()` 列举目标文件夹内的所有文件,并筛选出扩展名为 `.npy` 的文件[^2]。接着逐一调用 `np.load()` 将其内容加载到内存中并存入列表 `data_list` 中。
#### 使用 glob 模块简化操作
如果希望减少手动拼接字符串的工作量,则可采用 `glob` 模块来完成相同功能[^3]:
```python
import glob
import numpy as np
pattern = "./path/to/npy/files/*.npy"
file_paths = glob.glob(pattern)
loaded_data = []
for path in file_paths:
array = np.load(path)
loaded_data.append(array)
if loaded_data:
print("已加载数据:", len(loaded_data))
else:
print("未找到任何匹配的 .npy 文件")
```
此版本直接指定了通配符模式 `"*.npy"` 并由 `glob.glob()` 返回符合条件的所有绝对路径集合[^4]。
---
python保存npy文件
可以使用numpy库中的save函数来保存npy文件,例如:
import numpy as np
data = np.array([1, 2, 3, 4, 5])
np.save('data.npy', data)
这样就可以将data数组保存为data.npy文件。如果需要读取npy文件,可以使用numpy库中的load函数,例如:
data = np.load('data.npy')
print(data)
输出结果为:[1 2 3 4 5]
阅读全文
相关推荐
















