【解决问题】raise Exception(“Data must be 1-dimensional“) Exception: Data must be 1-dimensional
根据您提供的错误信息,问题出在尝试将y_test
转换为Pandas DataFrame时。错误提示“Data must be 1-dimensional”表明y_test
不是一个一维数组,而Pandas的DataFrame要求每个列是一维的。
您需要确保在创建DataFrame之前,y_test
是一个一维数组。通常,在读取CSV文件后,即使数据只有一列,Pandas也会将其作为二维数组处理。可以通过添加.squeeze()
方法来解决这个问题,这将把二维数组转换为一维数组(如果可能)。
修改代码如下:
# 获取数据集
x_test, y_test = data_preprocess()
# 确保y_test是一维数组
y_test = y_test.squeeze()
y_test_len = len(y_test)
# 创建一个DataFrame
data = {'index': range(y_test_len), 'label': y_test}
df = pd.DataFrame(data)
# 保存为CSV文件,包含表头
csv_file_path = 'y_test_labels_true.csv' # 指定保存路径
df.to_csv(csv_file_path, index=False)
print('CSV文件保存成功')
在这个修改中,.squeeze()
方法将y_test
数组从可能的二维形式(例如,(n, 1))转换为一维形式((n,)),这样就可以正确地创建DataFrame并保存为CSV文件了。