BAnet/BAset.py at master · SinLaber/BAnet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import random
import networkx as nx
import matplotlib.pyplot as plt


def ba_bip_graph(n):
    g = nx.DiGraph()
    f = open("start.txt")
    line = f.readline()
    repeated_user_nodes = []
    repeated_item_nodes = []
    while line:
        repeated_user_nodes.append(line.strip('\n').split(" ")[0])
        repeated_item_nodes.append(line.strip('\n').split(" ")[1])
        line = f.readline()
    users = 5
    user_line = 0
    item_line = 0
    while users < n:
        m1 = random.randint(1, 10)                                   #新增一个user，添加随机0-9个边
        # print("m1:", m1)
        item_target = set()
        while len(item_target) < m1:
            x = random.choice(repeated_item_nodes)
            item_target.add(x)
        label = []
        f = open("user_name.txt")
        line = f.readline()
        while line:
            label.append(line.strip('\n').split(" ")[0])
            line = f.readline()
        # print([label[user_line]] * m1)
        # print(item_target)
        g.add_edges_from(zip([label[user_line]]*m1, item_target))
        repeated_item_nodes.extend(item_target)
        repeated_user_nodes.extend([label[user_line]]*m1)
        user_line += 1
        f.close()
        #新增一个user，新增4个item
        repeat = 0
        while repeat < 4:
            m2 = random.randint(1, 5)
            # print("m2:", m2)
            user_target = set()
            while len(user_target) < m2:
                x = random.choice(repeated_user_nodes)
                user_target.add(x)
            label = []
            f_item = open("item_name.txt")
            line = f_item.readline()
            while line:
                label.append(line.strip('\n').split(" ")[0])
                line = f_item.readline()
            g.add_edges_from(zip([label[item_line]]*m2, user_target))
            repeated_user_nodes.extend(user_target)
            repeated_item_nodes.extend([label[item_line]] * m2)
            item_line += 1
            f.close()
            repeat += 1
        users = users+1
    return g


BA = ba_bip_graph(10)
pos = nx.circular_layout(BA)
nx.draw(BA, pos, with_labels=True, node_size=20)
plt.show()
for edge in BA.edges:
    print(edge)