把三个npz文件按相同的key合并成一个npz
时间: 2024-05-07 13:20:46 浏览: 247
Onekey 单文件版本
可以使用numpy库中的`np.load`和`np.savez`函数来实现。假设三个npz文件分别为`file1.npz`、`file2.npz`和`file3.npz`,以下是一个合并它们的示例代码:
```python
import numpy as np
# 加载三个npz文件
data1 = np.load('file1.npz')
data2 = np.load('file2.npz')
data3 = np.load('file3.npz')
# 合并数据
merged_data = {}
for key in data1.keys():
merged_data[key] = np.concatenate([data1[key], data2[key], data3[key]], axis=0)
# 保存合并后的数据
np.savez('merged_data.npz', **merged_data)
```
这个代码将会把三个npz文件中所有相同的key对应的数组按行合并成一个新的npz文件`merged_data.npz`。如果三个npz文件中有key不同的数组,那么这些数组将不会被合并。
阅读全文