如何为numpy矩阵生成干净的x轴和y轴?
问题描述:
我创建在numpy的距离矩阵,用out把这样:如何为numpy矩阵生成干净的x轴和y轴?
['H', 'B', 'D', 'A', 'I', 'C', 'F']
[[ 0. 2.4 6.1 3.2 5.2 3.9 7.1]
[ 2.4 0. 4.1 1.2 3.2 1.9 5.1]
[ 6.1 4.1 0. 3.1 6.9 2.8 5.2]
[ 3.2 1.2 3.1 0. 4. 0.9 4.1]
[ 5.2 3.2 6.9 4. 0. 4.7 7.9]
[ 3.9 1.9 2.8 0.9 4.7 0. 3.8]
[ 7.1 5.1 5.2 4.1 7.9 3.8 0. ]]
我打印的是x轴仅通过打印列表来之前,我打印的实际矩阵,:
print" ", names
print a
我需要该轴的顺序,因为列表'名称'正确地将变量与它们在矩阵中的值排序。但是,我将如何能够在numpy中获得类似的y轴?
答
也不是那么漂亮,但这个漂亮的表格打印的工作原理:
import numpy as np
names=np.array(['H', 'B', 'D', 'A', 'I', 'C', 'F'])
a=np.array([[ 0., 2.4, 6.1, 3.2, 5.2, 3.9, 7.1],
[2.4, 0., 4.1, 1.2, 3.2, 1.9, 5.1],
[6.1, 4.1, 0., 3.1, 6.9, 2.8, 5.2],
[3.2, 1.2, 3.1, 0., 4., 0.9, 4.1],
[5.2, 3.2, 6.9, 4., 0., 4.7, 7.9],
[3.9, 1.9 , 2.8, 0.9, 4.7, 0., 3.8],
[7.1, 5.1, 5.2, 4.1, 7.9, 3.8, 0. ]])
def pptable(x_axis,y_axis,table):
def format_field(field, fmt='{:,.2f}'):
if type(field) is str: return field
if type(field) is tuple: return field[1].format(field[0])
return fmt.format(field)
def get_max_col_w(table, index):
return max([len(format_field(row[index])) for row in table])
for i,l in enumerate(table):
l.insert(0,y_axis[i])
x_axis.insert(0,' ')
table.insert(0,x_axis)
col_paddings=[get_max_col_w(table, i) for i in range(len(table[0]))]
for i,row in enumerate(table):
# left col
row_tab=[str(row[0]).ljust(col_paddings[0])]
# rest of the cols
row_tab+=[format_field(row[j]).rjust(col_paddings[j])
for j in range(1,len(row))]
print(' '.join(row_tab))
x_axis=['x{}'.format(c) for c in names]
y_axis=['y{}'.format(c) for c in names]
pptable(x_axis,y_axis,a.tolist())
打印:
xH xB xD xA xI xC xF
yH 0.00 2.40 6.10 3.20 5.20 3.90 7.10
yB 2.40 0.00 4.10 1.20 3.20 1.90 5.10
yD 6.10 4.10 0.00 3.10 6.90 2.80 5.20
yA 3.20 1.20 3.10 0.00 4.00 0.90 4.10
yI 5.20 3.20 6.90 4.00 0.00 4.70 7.90
yC 3.90 1.90 2.80 0.90 4.70 0.00 3.80
yF 7.10 5.10 5.20 4.10 7.90 3.80 0.00
如果你想在X和Y轴是相同的,只是把它两个相同标签的列表。
+0
谢谢你,完全按照需要工作! – Sean 2012-08-01 15:35:38
你是什么意思?你想要在y方向上打印数组左侧的标识符(如电子表格)吗?或者你只是想在上面打印y标识符并在其下面打印数组的转置(因为数组/矩阵是对称的,在这种情况下,它会显示相同的内容)? – Evert 2012-08-01 13:59:28
我想打印数组左边的标识符,沿着y方向,就像我在数组顶部的x方向一样。使用包含我需要放置的特定订单的列表'名称'。 – Sean 2012-08-01 14:14:35
是的,x和y标识符都是相同的,因为它是一个对称矩阵,所以它基本上是一个反射。 – Sean 2012-08-01 14:15:16