在Python中：如何将稀疏矩阵写入csv文件？

Question

我是 Python 新手。我得到了一个脚本，它生成随机数并将它们放入 126x81 稀疏矩阵中。我想生成一个 csv 文件：Cell_ID；细胞_X；细胞_Y；瓦尔。单元格 X 和 Y 当然是每个单元格的坐标。我的脚本有一个循环，它生成一个“outputs.csv”文件，但其中的数据没有按照我想要的方式显示（每行的开头/结尾有方括号，并且有省略号 [... ] 代替某些值）。总而言之，我无法阅读矩阵的全部内容。

如果可以的话，我会上传一张图片，向您展示我希望这些数据在记事本或 Excel 中读取时的样子，但我不被允许这样做。然而，这些数据应该看起来像一个典型的 csv 文件，每个值都在其列下对齐。谢谢你们的帮助！:)

Answer 1

由于标准库支持 csv，所以我们使用它：

import numpy
import csv

N = 126
M = 81

g = numpy.random.rand(N, M)

with open('test.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(['x', 'y', 'value'])
    for (n, m), val in numpy.ndenumerate(g):
        writer.writerow([n, m, val])