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=',')