optimized_percolation.py

import numpy as np
import matplotlib.pyplot as plt

def label(Lw, p):
    R = np.random.rand(Lw, Lw) <= p
    label = np.zeros((Lw, Lw), dtype=int)
    current_label = 1
    
    for i in range(Lw):
        for j in range(Lw):
            if R[i, j]:
                above = label[i-1, j] if i > 0 else 0
                left = label[i, j-1] if j > 0 else 0

                if above == 0 and left == 0:
                    label[i, j] = current_label
                    current_label += 1
                elif above != 0 and left == 0:
                    label[i, j] = above
                elif above == 0 and left != 0:
                    label[i, j] = left
                else:
                    label = find_and_replace(label, i, j, above, left)
                    label[i, j] = left

    return label

def find_and_replace(label, i, j, above, left):
    row, col = np.where(label == above)
    for r, c in zip(row, col):
        label[r, c] = left
    return label

def main():
    Lw = 34
    p = 0.6
    L = label(Lw, p)
    print(L)
    plt.imshow(L)
    plt.colorbar()
    plt.show()

if __name__ == "__main__":
    main()