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

Question

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

我想用python编辑这个CSV，让矩阵的对角线元素替换为1，怎么办？</p>

我在 Windows 8.1 x64 下使用 python 2.7.4。

Answer 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)