Fonctions variadiques (...)

Fait partie de la section Logique et flux du Journey Lua de Coddy. Leçon 11 sur 54.

Parfois, vous avez besoin d'une fonction capable de travailler avec n'importe quel nombre d'arguments. Peut-être voulez-vous calculer la somme de deux nombres, ou trois, ou dix — sans écrire de fonctions distinctes pour chaque cas. Lua propose une syntaxe spéciale pour cela : les trois points ...

Lorsque vous placez ... dans la liste des paramètres d'une fonction, elle devient une fonction variadique qui accepte n'importe quel nombre d'arguments :

function sum(...)
    -- Corps de la fonction
end

À l'intérieur de la fonction, vous pouvez accéder à ces arguments en utilisant à nouveau .... Pour travailler avec eux plus facilement, vous pouvez les regrouper dans une table en utilisant des accolades :

function sum(...)
    local args = {...}
    local total = 0
    for i = 1, #args do
        total = total + args[i]
    end
    return total
end

print(sum(5, 10, 15))  -- Résultat : 30
print(sum(1, 2, 3, 4, 5))  -- Résultat : 15

L'expression {...} crée une table de type liste contenant tous les arguments passés à la fonction. Vous pouvez ensuite parcourir cette table comme n'importe quelle autre, en utilisant l'opérateur de longueur # ou ipairs().

Défi

Facile

Écrivez une fonction calculateAverage qui accepte n'importe quel nombre d'arguments numériques et retourne leur moyenne.

Utilisez la syntaxe variadique ... pour accepter un nombre flexible d'arguments, emballez-les dans une table, calculez leur somme et divisez par le nombre d'éléments.

Note : Le code de départ (marqué "Don't touch this code") lit des nombres à partir de l'entrée, convertit chacun d'eux du texte en nombre à l'aide de tonumber(), et les stocke dans une table nommée numbers. À la fin, il appelle votre fonction en utilisant unpack(numbers), ce qui déploie les valeurs de la table comme des arguments individuels — ainsi votre fonction les reçoit sous la forme .... Vous n'avez pas besoin de modifier ou de comprendre ce code en détail ; concentrez-vous simplement sur l'écriture de calculateAverage.

Logique :

Emballez tous les arguments dans une table en utilisant {...}
Parcourez la table pour calculer la somme de tous les nombres
Divisez la somme par le nombre d'arguments pour obtenir la moyenne
Retournez la moyenne sous forme de nombre

Paramètres :

... (nombres) : Nombre variable d'arguments numériques

Retourne : La moyenne de tous les nombres fournis (nombre)

Essayez vous-même

-- Ne pas toucher à ce code --
local numbers = {}
while true do
    local input = io.read()
    if input == nil or input == "" then
        break
    end
    local num = tonumber(input)
    if num then
        table.insert(numbers, num)
    end
end
-------------------------

function calculateAverage(...)
    -- Écrire le code ici
end

local average = calculateAverage(unpack(numbers))
print("Average: " .. average)