在Python中创建列表中的连续项矩阵

问题描述 投票:0回答:1

我有一个玩家列表,我试图找到这些玩家成对出现的矩阵。

players = [Player52, Player48,Player41,Player42,Player52,Player48]
matrix = 
          Player52  Player48  Player41  Player42
Player52      0         2        0          0
Player48      0         0        1          0
Player41      0         0        0          1
Player42      1         0        0          0

我尝试制作一个源,目标和权重列表来形成一个数据框,并使用邻接矩阵的概念来创建我想要的矩阵。

import json
import pandas as pd
import numpy as np
with open('interactions_clean.json') as f:
    data = json.load(f)

for i in data:
    player_list.append(i['dataItem']['Name'])

first = []
second =[]
for i in range(len(players) - 1):
    first.append(players[i])
    second.append(players[i+1])

weight = [1]*684

matrix_ad = pd.DataFrame({'source': first, "target": second, 'weight': weight})
vals = np.unique(matrix_ad[['source', 'target']])

所以为了创建我想要的矩阵,我尝试了几件事,比如

x= matrix_ad.pivot_table(index='source',columns='target',values='weight')

这确实给了我一个矩阵但是没有添加重复对的出现。

python-3.x matrix adjacency-matrix
1个回答
0
投票

删除唯一的行:

vals = np.unique(matrix_ad[['source', 'target']])

应该是好的。您不想删除那里的重复项。

© www.soinside.com 2019 - 2024. All rights reserved.