Python矩陣是很常見的一種數(shù)據(jù)結(jié)構(gòu),并且在很多數(shù)據(jù)科學領(lǐng)域都得到了廣泛的應用。在實際的項目中,我們可能需要將矩陣數(shù)據(jù)寫入文件中進行存儲,Python提供了非常方便的方法。
# 創(chuàng)建一個3*3的矩陣
import numpy as np
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 打開一個文件,將矩陣數(shù)據(jù)寫入文件中
with open('matrix.txt', 'w') as f:
for row in matrix:
f.write(' '.join(str(num) for num in row))
f.write('\n')
以上代碼會將創(chuàng)建的矩陣寫入到當前工作目錄下的`matrix.txt`文件中,每一行都是一個數(shù)組的數(shù)據(jù)。如果想要在讀取時將數(shù)據(jù)還原成矩陣的形式,可以使用以下代碼:
# 打開文件,讀取矩陣數(shù)據(jù)
with open('matrix.txt', 'r') as f:
lines = f.readlines()
# 將每一行數(shù)據(jù)轉(zhuǎn)換為數(shù)組,然后拼接成一個矩陣
matrix = np.array([list(map(int, line.strip().split())) for line in lines])
以上代碼會將之前寫入的矩陣數(shù)據(jù)讀取出來,并且還原成`numpy`數(shù)組的形式,方便之后的使用。