在Python中：如何將稀疏矩陣寫入CSV文件？

Question

我是Python的新手。 我得到了一個腳本，該腳本生成隨機數並將其放入126x81稀疏矩陣中。 我想使用以下代碼生成一個csv文件： Cell_ID ; Cell_X ; Cell_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])