plot a 2d self avoiding random walk in python

main.py
import numpy as np
import matplotlib.pyplot as plt
 
# Function to generate a 2D self-avoiding random walk
def self_avoiding_RW(n):
    positions = np.zeros((n, 2))
    positions[0] = [0, 0]
 
    for i in range(1, n):
        dx, dy = 0, 0
        while dx == 0 and dy == 0:
            dx, dy = np.random.choice([-1, 1]), np.random.choice([-1, 1])
        new_pos = positions[i-1] + [dx, dy]
        while any((new_pos == p).all() for p in positions[:i]):
            dx, dy = 0, 0
            while dx == 0 and dy == 0:
                dx, dy = np.random.choice([-1, 1]), np.random.choice([-1, 1])
            new_pos = positions[i-1] + [dx, dy]
        positions[i] = new_pos
    return positions
 
# Plotting the walk
positions = self_avoiding_RW(100)
plt.plot(positions[:, 0], positions[:, 1], 'b')
plt.show()
816 chars
26 lines