23 lines
734 B
Python
23 lines
734 B
Python
from pyvis.network import Network
|
|
import pyvis
|
|
from random import randint
|
|
|
|
|
|
pyvis.__version__
|
|
|
|
|
|
def random_colour():
|
|
return f"rgb({randint(0,255)},{randint(0,255)},{randint(0,255)})"
|
|
|
|
def main():
|
|
net = Network(notebook=False, cdn_resources="remote", neighborhood_highlight=True, bgcolor="#222222", font_color="#FAFAFA")
|
|
net.repulsion(spring_length=300)
|
|
|
|
net.add_nodes([1,2,3,4,5,6,7,8,9,10], label=["1","2","3","4","5","6","7","8","9","10"],color=[random_colour() for _ in range(10)])
|
|
net.add_edges([(1, 2, 10), (1, 3, 5), (1, 4, 6), (2, 3, 10), (2, 5, 1), (5, 6, 9), (7, 8, 7), (8, 9, 7), (9, 7, 8), (1, 7, 7), (9, 10, 2)])
|
|
|
|
net.show("network.html", notebook=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |