python 矩阵类型转换_Python-将矩阵转换为边列表/长格式

您还需要枚举列标题,才能打印出各个单元格.

对于矩阵文件mat.csv：

null,A,B,C

A,0,2,3

B,3,4,2

C,1,2,4

以下程序：

csv = open("mat.csv")

columns = csv.readline().strip().split(',')[1:]

for line in csv:

tokens = line.strip().split(',')

row = tokens[0]

for column, cell in zip(columns,tokens[1:]):

print '{},{},{}'.format(row,column,cell)

打印出：

A,A,0

A,B,2

A,C,3

B,A,3

B,B,4

B,C,2

C,A,1

C,B,2

C,C,4

要生成上对角线,可以使用以下脚本：

csv = open("mat.csv")

columns = csv.readline().strip().split(',')[1:]

for i, line in enumerate(csv):

tokens = line.strip().split(',')

row = tokens[0]

for column, cell in zip(columns[i:],tokens[i+1:]):

print '{},{},{}'.format(row,column,cell)

结果为：

A,A,0

A,B,2

A,C,3

B,B,4

B,C,2

C,C,4