Python numpy矩阵加列名并合并数组

最新推荐文章于 2024-08-17 03:41:52 发布

原创最新推荐文章于 2024-08-17 03:41:52 发布 · 8.2k 阅读

2 ·

CC 4.0 BY-SA版权

数据挖掘专栏收录该内容

7 篇文章

订阅专栏

本文介绍如何在预测结果y_pred上添加列名，确保数据的完整性。通过使用pandas库，将预测结果与有列名的数据X进行合并，并保存为CSV文件。

y_pred是100*1的数组，本文主要是在y_pred上方加一行类别列名

注意：不识别中文列名
前面加入

#-*- coding : utf-8 -*-
# coding: utf-8

X为有列名的数据在这里插入图片描述

 listHaveRow = pd.read_csv(X,header=None,error_bad_lines=False,encoding="gbk")
    listHaveRow=np.array(listHaveRow)

必须添加header=None，否则默认把第一行数据处理成列名删除导致缺失

  # 保存
    matr="类别"
    y_predName = []
    y_predName.append(matr)
    y_predName=np.array(y_predName)

    matrAll=np.concatenate((y_predName,y_pred),axis=0)
    matrAll=np.column_stack((listHaveRow,matrAll))
    # print(matrAll)
    # 字典中的key值即为csv中列名
    dataframe = pd.DataFrame(data=matrAll)

    # 将DataFrame存储为csv,index表示是否显示行名，default=True
    dataframe.to_csv(savePath, header=False, index=False, sep=',')