Salut les kheys,

Je débute en dev j'essaie de générer des valeurs aléatoires dans pyspark et de les écrire dans un fichier JSON. Je génére 100 enregistrements dans mon code mais dans mon fichier JSON j'ai qu'un enregistrement. J'ai un petit soucis lors de l'incrémentation, voici mon code :

import random
import json


fruit = ['banana', 'apple', 'orange']
origine = ['france', 'sweden', 'USA', 'mexico']

def generate():
    for n in range(100):
       ID = n+1
       types_fruit = (random.choice(fruit))
       origine_names = (random.choice(origine))
       return (types_fruit+origine_names)

with open("sales_detail.json", "w") as sales : 
   json.dump(generate(), sales)

Merci pour votre aide

Prends ce que je te dis avec des pincettes, je suis sur mobile et le formattage est weird.

Dans ta fonction generate la variable ID n’est pas utilisé. Si tu ne l’utilise pas, ne la déclare pas
Ensuite tu redéfini tes valeurs à chaque itérations, mais tu return en fin de loop, donc à la fin de la première itération. -> tu auras un str qui sera composé d’un Type et d’une origine de fruit.
Beaucoup de tes parenthèses sont redondantes, tu pourrais retirer les parenthèses qui ne servent pas à call des fonction. Précise les si tu fais des calculs pour les priorités d’exécution et éviter les ambigüité.

 import random
import json


fruit = ['banana', 'apple', 'orange']
origine = ['france', 'sweden', 'USA', 'mexico']

def generate():
    for n in range(100):
       types_fruit = random.choice(fruit)
       origine_names = random.choice(origine)
       yield types_fruit+origine_names

with open("sales_detail.json", "w") as sales : 
   json.dump(generate(), sales) 

Test pour voir si dump accepte un generator sinon remplace l’appel de generate par ‘list(generate())’.

Juste que cette solution te fais créer un générateur (fonction avec yield).
C’est peut être avancé comme concept, si tu préfère tu peux ajouter une variable de type liste dans ta fonction et y append les valeurs. Ensuite après la loop tu return cette liste.