【发布时间】：2017-11-03 09:44:38
【问题描述】：
我尝试做的是将混淆矩阵保存在某种文本文件中（可能 HTML 看起来最好）。我需要从 csv 文件中获取数据，将其添加到数组中，然后创建混淆矩阵。代码：
import csv
import pandas as pd
data = csv.reader(open('results_date.csv', 'r'), delimiter=";", quotechar='|')
next(data)
true_data = []
pred_data = []
for row in data:
if len(row) >= 2:
true_data.append(row[0])
pred_data.append(row[1])
true_data = [s.strip().split('_')[0] for s in true_data]
pred_data = [s.strip().split('=')[0] for s in pred_data]
y_true = pd.Series(true_data, name="Actual")
y_pred = pd.Series(pred_data, name="Predicted")
df_confusion = pd.crosstab(y_true, y_pred)
print (df_confusion)
混淆矩阵如下所示：
Predicted class1 class2 class3 class4 classX
Actual
class1 5 6 0 4 5
class2 1 0 4 8 0
class3 5 3 2 0 1
class4 4 2 5 2 0
classX 0 5 2 1 7
我想以相同的形式将它输出到某个文件中（我的猜测是它作为 HTML 或 CSV 文件看起来会更好，但任何事情都可以 - 请不要使用需要特殊程序的疯狂格式） .
【问题讨论】：
标签： python html csv confusion-matrix