【Python】将矩阵、字典和张量存到txt文件中去

 一、将矩阵存储到txt文件中

import numpy as np
np.savetxt(r'A.txt', B, fmt='%s', delimiter=',') 

A.txt---------存储的txt文件名称

B-------------要存储的矩阵

举例：

import numpy as np


Test = np.array([[1, 7], [5, 3], [0, 1]])
np.savetxt(r'Test.txt', Test, fmt='%s', delimiter=',')  # 存储矩阵Test

Test.txt文件中内容

 二、将list of dict存储到txt文件

# 将数据list of dict保存到txt
test = [{1: [7, 7], 5: [3], 0: [1, 4], 2: [3]}]
with open('F:\Five_view\Tr_neg.txt', 'w+') as f:
    for i in range(0, len(test)):
        for k, v in test[i].items():
            f.write(f"{k}, {v}\n".format(k, v))

其中，F:\Five_view\Tr_neg.txt为存储文件位置

 三、将tensor存储到txt文件 

result1 = np.array(test_data)

np.savetxt("test_data.txt", result1)

其中，test_data为tensor类型