Question

如何用python画好confusion matrix

Answer 1

def is_tri_matrix(matrix):
size = len(matrix)
d = 'notsure'
for y in range(size):
for x in range(size):
if matrix[y][x] != 0 and x >= 0 and y <= x and d == 'notsure':
d = 'up'
elif matrix[y][x] != 0 and x <= y and y >= 0 and d == 'notsure':
d = 'down'
elif d == 'up' and matrix[y][x] != 0 and x >= 0 and y > x:
return False
elif d == 'down' and matrix[y][x] != 0 and x >=0 and y < x:
return False
elif y == x and matrix[y][x] != matrix[0][0]:
return False;
return True

'''
1 1 1
0 1 1
0 0 1
'''
m = [[1, 1, 1], [0, 1, 1], [0 ,0, 1]]
print is_tri_matrix(m)

'''
1 0 0
0 1 0
0 0 1
'''
n = [[1, 0, 0], [0, 1, 0], [0, 0, 1]]
print is_tri_matrix(n)

'''
1 0 1
0 1 0
1 0 1
'''
o = [[1, 0, 1], [0, 1, 0], [1, 0, 1]]
print is_tri_matrix(o)

python 2..7.6