处理pkl文件，并将结果映射到另一个二维矩阵中

方法一：

import pickle
import numpy as np
import pandas as pd
import itertools
import time

#读取pkl文件
with open(r"D:\python_learn\numpy_matrix.pkl", 'rb') as fo:
    data = pickle.load(fo, encoding='bytes')
    
datacopy = np.zeros((4000, 4000))#生成一个全0的二维矩阵
start =time.time()
for i in range(len(data)):
    index=np.where(data[i]==1)#找到每一行中数值为1元素的列坐标，type=nparray
    #print(index)
    index_list=index[0].tolist()#转化为列表
    #print(index_list)
    position = list(itertools.permutations(index_list, 2))#list中的元素两两排列，eg.(1,2),(2,1)
    for elem in position:
        datacopy[elem[0],elem[1]]+=1#将上一步得到的排列序号投影到该二维矩阵中，对应元素+1
        
np.savetxt(r"D:\python_learn\numpy_data.pkl",datacopy ,fmt="%d")#将结果保存为pkl文件

end = time.time()
print(f" 1耗时{end-start}") 

方法二：

import pickle
import numpy as np
import pandas as pd
from itertools import permutations
import time

with open(r"D:\python_learn\numpy_matrix.pkl", 'rb') as fo:
    data = pickle.load(fo, encoding='bytes')
    
total=np.sum(data, axis=0)#计算每一列的和    
datacopy = np.zeros((4000, 4000))

start =time.time()
for i in range(len(data)):
    index=np.where(data[i]==1)#获取元素为1的列位置索引
    index_list=index[0].tolist()
    for j in index_list:
        for k in index_list:#将内容为1的索引组成数对，映射到另一个矩阵中
            datacopy[j,k]+=1
            
np.savetxt(r"D:\python_learn\numpy_data.pkl",datacopy ,fmt="%d")

end = time.time()
print(f" 2耗时{end-start}")