failure.py

import networkx as nx
import random
import numpy as np

for k in range(5,64,10):
    for m in range(1000,10001,2000):
        mean_time_between_failure = m # unit is hour
        failRate = 1/mean_time_between_failure
        time_quantum = 1 # hour

        btree = nx.balanced_tree(k-1,h)
        nodelist = btree.nodes()
        # del nodelist[0]

        record = list()

        for t in range(365):
            temp = btree.copy()
            fails = sum(np.random.poisson(failRate,24))
            RandomSample = random.sample(nodelist, fails)
            if 0 in RandomSample:
                record.append(0)
                continue
            temp.remove_nodes_from(RandomSample)
            dtree = nx.dfs_tree(temp,0)
            record.append(dtree.number_of_nodes()/btree.number_of_nodes())

        print("{},{},{},{}".format(
            h,k,mean_time_between_failure,np.average(record)))