将npz文件中不同的key合并到一起

可以使用NumPy中的`load`函数逐个将不同的key加载到内存中，然后使用`concatenate`函数合并它们。以下是一个示例代码：

import numpy as np

# 加载第一个npz文件
data = np.load('file1.npz')
keys = list(data.keys()) # 获取所有的key

# 分别加载每个key的数据
arrs = []
for key in keys:
    arrs.append(data[key])

# 合并所有数据
merged_arr = np.concatenate(arrs, axis=0)

# 逐个加载其他npz文件，并重复以上操作
for i in range(2, num_files + 1):
    data = np.load(f'file{i}.npz')
    for key in keys:
        arrs.append(data[key])
    merged_arr = np.concatenate(arrs, axis=0)

请注意，此代码假定所有npz文件都具有相同的key。如果在某个文件中缺少特定的key，则会引发错误。如果您想要处理这种情况，请添加一些额外的代码以检查每个文件中的key是否存在，并根据需要跳过它们。

相关问题

将npz文件中6个不同的key，两两合并到一起

可以使用NumPy的load函数加载npz文件，然后使用NumPy的concatenate函数将不同的数组合并起来。以下是一个示例代码：

import numpy as np

# 加载npz文件
data = np.load('data.npz')

# 提取6个数组
arr1 = data['arr_0']
arr2 = data['arr_1']
arr3 = data['arr_2']
arr4 = data['arr_3']
arr5 = data['arr_4']
arr6 = data['arr_5']

# 合并数组
merged_arr1_2 = np.concatenate((arr1, arr2), axis=0)
merged_arr1_3 = np.concatenate((arr1, arr3), axis=0)
merged_arr1_4 = np.concatenate((arr1, arr4), axis=0)
merged_arr1_5 = np.concatenate((arr1, arr5), axis=0)
merged_arr1_6 = np.concatenate((arr1, arr6), axis=0)
merged_arr2_3 = np.concatenate((arr2, arr3), axis=0)
merged_arr2_4 = np.concatenate((arr2, arr4), axis=0)
merged_arr2_5 = np.concatenate((arr2, arr5), axis=0)
merged_arr2_6 = np.concatenate((arr2, arr6), axis=0)
merged_arr3_4 = np.concatenate((arr3, arr4), axis=0)
merged_arr3_5 = np.concatenate((arr3, arr5), axis=0)
merged_arr3_6 = np.concatenate((arr3, arr6), axis=0)
merged_arr4_5 = np.concatenate((arr4, arr5), axis=0)
merged_arr4_6 = np.concatenate((arr4, arr6), axis=0)
merged_arr5_6 = np.concatenate((arr5, arr6), axis=0)

其中，`data['arr_x']` 表示npz文件中的第 x 个数组。`np.concatenate` 函数的 `axis` 参数表示合并的方向，0表示纵向合并，1表示横向合并。以上代码将6个数组两两合并，共产生15个新数组。

把三个npz文件按相同的key合并成一个npz

可以使用numpy库中的`np.load`和`np.savez`函数来实现。假设三个npz文件分别为`file1.npz`、`file2.npz`和`file3.npz`，以下是一个合并它们的示例代码：

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不同的数组，那么这些数组将不会被合并。