python CSV编辑矩阵对角线中的元素

时间:2014-04-19 03:08:12

标签: python csv matrix diagonal

我有一个包含n * n距离矩阵的CSV文件,这里是视图的一部分,

enter image description here

我想使用python编辑这个CSV,使矩阵对角线中的元素替换为1,  我该怎么办?

我在Windows 8.1 x64下使用python 2.7.4。

1 个答案:

答案 0 :(得分:1)

最好从旧的csv创建一个新的csv。 (另)

import csv

newlines = []
with open('csvfilename.csv', 'rb') as csvfile:
  reader = csv.reader(csvfile)
  i=0
  for row in reader:
    newlines.append(row[:i] + [1] + row[i+1:])
    i+=1
with open('newcsvfilename.csv','wb') as csvfile:
  writer = csv.writer(csvfile)
  writer.writerows(newlines)