读取一个文件下的所有numpy数组
时间: 2023-12-18 18:05:08 浏览: 98
取numpy数组的某几行某几列方法
5星 · 资源好评率100%
以下是读取一个文件下所有numpy数组的Python代码:
```python
import numpy as np
import os
# 指定文件夹路径
folder_path = "path/to/folder"
# 遍历文件夹中每个文件
for file_name in os.listdir(folder_path):
file_path = os.path.join(folder_path, file_name)
# 判断是否为.npy文件
if file_name.endswith(".npy"):
# 读取numpy数组
np_array = np.load(file_path)
# 处理numpy数组
# ...
```
这段代码首先指定了一个文件夹路径,然后遍历该文件夹中的每个文件。对于每个文件,判断其是否为.npy文件,并使用`np.load()`函数读取numpy数组。可以在读取完数组后进行相应的处理。
阅读全文