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

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

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

---

我正在Windows 8.1 x64下使用Python2.7.4。

最好从旧的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)