python编码实现读取文件夹下的所有npy文件,将读取的数据相加求和
时间: 2023-07-18 12:16:11 浏览: 155
好的,以下是 Python 代码实现:
```python
import numpy as np
import os
folder_path = "your_folder_path" # 文件夹路径
sum_data = None # 初始化数据和为 None
for file_name in os.listdir(folder_path): # 遍历文件夹下的每个文件
if file_name.endswith(".npy"): # 判断文件是否是 npy 文件
file_path = os.path.join(folder_path, file_name) # 获取文件路径
data = np.load(file_path) # 加载文件数据
if sum_data is None: # 如果是第一个文件,将数据和初始化为当前文件数据
sum_data = data
else: # 如果不是第一个文件,将当前文件数据加到数据和上
sum_data += data
print("数据和为:", sum_data)
```
解释一下上面的代码:
1. 首先,我们需要指定文件夹的路径,可以使用 `folder_path` 变量来代替。
2. 然后,我们需要遍历文件夹下的每个文件,可以使用 `os.listdir()` 函数来获取文件夹下的文件列表。
3. 对于每个文件,我们需要判断它是否是 npy 文件,可以使用 `str.endswith()` 函数来判断。
4. 如果是 npy 文件,我们需要使用 `np.load()` 函数加载文件数据。
5. 如果是第一个文件,我们需要将数据和初始化为当前文件数据;否则,我们需要将当前文件数据加到数据和上。
6. 最后,我们输出数据和即可。
希望能够帮到你!