In this article, we'll discuss *args
and ****kwargs**
in Python along with their uses and some examples.
When writing a function, we often need to pass values to the function. These values are called function arguments.
Problem with Function Arguments
Let's define a function to add two numbers in Python. We'll write it like this:
def add(x, y):
return x+y
print(add(2,3))
Output:
5
What if you need to add three numbers? Simple, we can modify the function to accept three arguments and return their sum as:
def add(x, y, z):
return x+y+z
print(add(2, 3, 5))
Output:
10
Wasn't that quite simple? Yes, it was!
But what if we're again required to add two numbers only? Will our modified function help us get the sum? Let's see:
def add(x, y, z):
return x+y+z
print(add(2, 3))
Output:
Traceback (most recent call last):
File "D:\Quarantine\Test\Blog-Codes\args-kwargs\main.py", line 14, in <module>
print(add(2, 3))
TypeError: add() missing 1 required positional argument: 'z'
You see the problem?
The problem arises when we have a variable number of arguments. Should we keep modifying the function to accept the exact number of arguments? Of course not, we won't be doing this.
So there must be some other way to do it. This is where *args
and ****kwargs**
jump in.
You can use *args
and ****kwargs**
as arguments of a function when you are unsure about the number of arguments to pass in the functions.
How to Use *args in Python
*args
allows us to pass a variable number of non-keyword arguments to a Python function. In the function, we should use an asterisk (*
) before the parameter name to pass a variable number of arguments.
def add(*args):
print(args, type(args))
add(2, 3)
Output:
(2, 3) <class 'tuple'>
Thus, we're sure that these passed arguments make a tuple inside the function with the same name as the parameter excluding *
.
Now let's rewrite our **add()**
function with a variable number of arguments.
def add(*numbers):
total = 0
for num in numbers:
total += num
return total
print(add(2, 3))
print(add(2, 3, 5))
print(add(2, 3, 5, 7))
print(add(2, 3, 5, 7, 9))
Output:
5
10
17
26
Note that the name of the argument need not necessarily be **args**
– it can be anything. In this case it's **numbers**
. But it's generally a standard way to use ***args**
as the name.
How to Use **kwargs in Python
`kwargs** allows us to pass a variable number of keyword arguments to a Python function. In the function, we use the double-asterisk (**
`) before the parameter name to denote this type of argument.
def total_fruits(**kwargs):
print(kwargs, type(kwargs))
total_fruits(banana=5, mango=7, apple=8)
Output:
{'banana': 5, 'mango': 7, 'apple': 8} <class 'dict'>
Thus we see that the arguments, in this case, are passed as a dictionary and these arguments make a dictionary inside the function with name same as the parameter excluding ``**.
Now, let's complete the **total_fruits()**
function to return the total number of fruit.
def total_fruits(**fruits):
total = 0
for amount in fruits.values():
total += amount
return total
print(total_fruits(banana=5, mango=7, apple=8))
print(total_fruits(banana=5, mango=7, apple=8, oranges=10))
print(total_fruits(banana=5, mango=7))
Output:
20
30
12
Note that the name of the argument need not necessarily be **kwargs**
– again, it can be anything. In this case, it's **fruits**
. But it's generally a standard way to use ****kwargs**
as the name.
Conclusion
In this article, we learned about two special keywords in Python – *args
and ****kwargs**
. These make a Python function flexible so it can accept a variable number of arguments and keyword arguments, respectively.
Thanks for reading!
You can find the code for this blog here.