How to return a list in Python?

Returning a List in Python: A Comprehensive Guide

Introduction

In Python, lists are a fundamental data structure that can be used to store and manipulate collections of data. When working with lists, you may need to return them from functions or methods. In this article, we will explore the different ways to return a list in Python, including how to create lists, append elements, and return lists from functions.

Creating Lists

Before we can return a list, we need to create one. In Python, we can create a list using the [] syntax. Here’s an example:

# Create a list

my_list = [1, 2, 3, 4, 5]

Append Elements

Once we have created a list, we can append elements to it using the append() method. This method adds an element to the end of the list.

# Append an element to the list

my_list.append(6)

print(my_list)  # Output: [1, 2, 3, 4, 5, 6]

Return a List

To return a list from a function, we need to use the return statement. Here’s an example:

# Define a function that returns a list

def get_list():

    return [1, 2, 3, 4, 5]



# Call the function and print the returned list

list_from_function = get_list()

print(list_from_function)  # Output: [1, 2, 3, 4, 5]

Return Multiple Elements

If we want to return multiple elements from a function, we can use a list comprehension or the return statement with multiple values.

# Define a function that returns multiple elements

def get_multiple_elements():

    return [1, 2, 3, 4, 5, 6]



# Call the function and print the returned list

list_from_function = get_multiple_elements()

print(list_from_function)  # Output: [1, 2, 3, 4, 5, 6]

Return a List of Lists

If we want to return a list of lists, we can use the return statement with a list comprehension.

# Define a function that returns a list of lists

def get_list_of_lists():

    return [[1, 2], [3, 4], [5, 6]]



# Call the function and print the returned list

list_from_function = get_list_of_lists()

print(list_from_function)  # Output: [[1, 2], [3, 4], [5, 6]]

Return a List of Tuples

If we want to return a list of tuples, we can use the return statement with a list comprehension.

# Define a function that returns a list of tuples

def get_list_of_tuples():

    return [(1, 2), (3, 4), (5, 6)]



# Call the function and print the returned list

list_from_function = get_list_of_tuples()

print(list_from_function)  # Output: [(1, 2), (3, 4), (5, 6)]

Return a List of Dictionaries

If we want to return a list of dictionaries, we can use the return statement with a list comprehension.

# Define a function that returns a list of dictionaries

def get_list_of_dictionaries():

    return [{'name': 'John', 'age': 30}, {'name': 'Jane', 'age': 25}, {'name': 'Bob', 'age': 40}]



# Call the function and print the returned list

list_from_function = get_list_of_dictionaries()

print(list_from_function)  # Output: [{'name': 'John', 'age': 30}, {'name': 'Jane', 'age': 25}, {'name': 'Bob', 'age': 40}]

Return a List of Sets

If we want to return a list of sets, we can use the return statement with a list comprehension.

# Define a function that returns a list of sets

def get_list_of_sets():

    return [{1, 2}, {3, 4}, {5, 6}]



# Call the function and print the returned list

list_from_function = get_list_of_sets()

print(list_from_function)  # Output: [{1, 2}, {3, 4}, {5, 6}]

Return a List of Mixed Data Types

If we want to return a list of mixed data types, we can use the return statement with a list comprehension.

# Define a function that returns a list of mixed data types

def get_list_of_mixed_data_types():

    return [1, 'hello', 3.14, True]



# Call the function and print the returned list

list_from_function = get_list_of_mixed_data_types()

print(list_from_function)  # Output: [1, 'hello', 3.14, True]

Return a List with Custom Type

If we want to return a list with a custom type, we can use the return statement with a list comprehension.

# Define a function that returns a list with a custom type

def get_list_with_custom_type():

    return [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]



# Call the function and print the returned list

list_from_function = get_list_with_custom_type()

print(list_from_function)  # Output: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

Return a List with Custom Class

If we want to return a list with a custom class, we can use the return statement with a list comprehension.

# Define a class

class Person:

    def __init__(self, name, age):

        self.name = name

        self.age = age



# Define a function that returns a list with a custom class

def get_list_with_custom_class():

    return [Person('John', 30), Person('Jane', 25), Person('Bob', 40)]



# Call the function and print the returned list

list_from_function = get_list_with_custom_class()

for person in list_from_function:

    print(person.name, person.age)

# Output:

# John 30

# Jane 25

# Bob 40

Return a List with Custom Type and Class

If we want to return a list with a custom type and class, we can use the return statement with a list comprehension.

# Define a class

class Person:

    def __init__(self, name, age):

        self.name = name

        self.age = age



# Define a function that returns a list with a custom type and class

def get_list_with_custom_type_and_class():

    return [Person('John', 30), Person('Jane', 25), Person('Bob', 40)]



# Call the function and print the returned list

list_from_function = get_list_with_custom_type_and_class()

for person in list_from_function:

    print(person.name, person.age)

# Output:

# John 30

# Jane 25

# Bob 40

Conclusion

In this article, we have explored the different ways to return a list in Python, including how to create lists, append elements, and return lists from functions. We have also discussed how to return multiple elements, return a list of lists, return a list of tuples, return a list of dictionaries, return a list of sets, return a list of mixed data types, return a list with a custom type, and return a list with a custom class. Finally, we have demonstrated how to return a list with a custom type and class.