节点（个体）级SIS模型python代码

原来做实验我用matlab写代码。
现在想用machine learning来做cyber security，改成python方便些。
网上一搜仓室级模型代码不少，但是节点（个体）级模型代码没有。
这里贡献一发，出图效果如下：

# -*- coding: utf-8 -*-
# @Time    : 2018/3/27 18:00
# @Author  : timothy

import scipy.integrate as spi
import numpy as np
import pylab as pl
import networkx as nx
import random as rd


# 初始化
BETA =