如果我的一个路径下有多个npy文件,他们中是维度相同的数组,怎么全部读取他们并累加
时间: 2024-05-05 20:17:45 浏览: 75
Numpy对数组的操作:创建、变形(升降维等)、计算、取值、复制、分割、合并
你可以使用NumPy库中的load函数来读取npy文件,并使用for循环遍历所有文件并累加数组。下面是一个示例代码:
```
import numpy as np
# 定义路径和文件名前缀
path = 'your_path/'
prefix = 'file_'
# 定义数组变量,用于累加
sum_array = None
# 遍历所有文件
for i in range(1, 10):
file_name = prefix + str(i) + '.npy'
file_path = path + file_name
# 读取npy文件
array = np.load(file_path)
# 累加数组
if sum_array is None:
sum_array = array
else:
sum_array += array
# 打印累加后的数组
print(sum_array)
```
这段代码假设路径为'your_path/',文件名前缀为'file_',有9个npy文件,分别为'file_1.npy'到'file_9.npy'。你可以根据实际情况修改路径和文件名前缀,并根据需要修改遍历的文件数。
阅读全文