import numpy as np
import pandas as pd
# 加载.npz文件,更换为自己的文件名
data = np.load('kcn-torch-master/datasets/bird_count.npz')
# 查看文件中的数组及其名称
print(data.keys())
# 假设我们想要将训练特征、训练标签、测试特征和测试标签转换为CSV
# 根据实际的键名进行调整
X_train = data['Xtrain']
Y_train = data['Ytrain']
X_test = data['Xtest']
Y_test = data['Ytest']
# 将NumPy数组转换为pandas DataFrame
df_train = pd.DataFrame(X_train)
df_train['label'] = Y_train.ravel() # 将标签添加到DataFrame
df_test = pd.DataFrame(X_test)
df_test['label'] = Y_test.ravel() # 将标签添加到DataFrame
# 保存为CSV文件
df_train.to_csv('bird_count_train.csv', index=False)
df_test.to_csv('bird_count_test.csv', index=False)
结果显示如下: