Bonjour je dois réaliser un travail dont le sujet est le suivant même si je ne pense pas qu'il sera utile pour répondre à ma question

Une fourmi parcourt de manière aléatoire les côtes d'un triangle équilatéral direct ABC. Sa position initiale sur l'un des sommets A, B ou C est décidée de manière aléatoire et équiprobable. Ensuite, quand elle se trouve sur un sommet elle se déplace vers l'un des deux autres sommets. La probabilité qu'elle se déplace dans le sens trigonométrique est 3/4.

Les sommets À, B et C du triangle sont repérés respectivement par les nombres 1, 2 et 3. L'algorithme incomplet suivant détermine le numéro du 2ème sommet fréquenté par la fourmi

J'ai réalisé ce programme Python, (mais je suis obligée d'utiliser une fonction (def return))

n,t=10,40
for j in range(1,n+1):
    p=randint(1,3)
    for i in range(1,t+1):
        a=random()
        if a<0.75:
            p=p+1
            if p==4:
                p=1
        else:
            p=p-1
            if p==0:
                p=3
        print (p,end="")
il marche très bien et les résultats sont bons
Afin d'utiliser obligatoirement une fonction avec def et return j'ai ainsi réalisé ce programme qui est plus court et synthétisé

from random import randint,random

def fourmi():
    n=100
    p=randint(1,3)
    for i in range(1,n+1):
        a=random()
        if a<0.75:
            p=p+1
            if p==4:
                p=1
        else:
            p=p-1
            if p==0:
                p=3
        return(p)

j'ai tout essayé aidez moi s'il vous plait
sincèrement merci pour votre attention