How To Store List In Array Python

In this tutorial, we will learn about storing lists in arrays using Python. Arrays can be used to store multiple values in one place, making it convenient to organize our data.

Python provides a built-in list data type that can be used to store multiple items in a single variable. We will also discuss the use of NumPy arrays, which offer more functionalities compared to lists and are specifically designed for numerical operations.

Step 1: Using Lists

A list can be created using square brackets, and items inside the list are separated by commas. To initialize a list, follow the syntax below:

1
my_list = [value1, value2, value3, ...]

For example, to store a list of integers:

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

Step 2: Accessing List Elements

To access elements in a list, you can use indexing. Indices start at 0 in Python. Here’s an example:

1
2
3
# Accessing the first element
first_element = integer_list[0]
second_element = integer_list[1]

Step 3: Using NumPy Arrays

Python’s built-in lists are great for general purposes, but when dealing with numerical data, it is more efficient to use a numeric library like NumPy. NumPy is a popular and powerful library for working with arrays in Python. To install NumPy, open your terminal or command prompt and type:

1
pip install numpy

To create an array in NumPy, first import the library and use the numpy.array() function as follows:

1
2
3
import numpy as np
 
numpy_array = np.array([1, 2, 3, 4, 5])

Step 4: Accessing Elements in NumPy Arrays

Accessing elements in NumPy arrays is similar to accessing elements in Python lists. However, NumPy provides more advanced slicing and indexing options. Let’s try accessing elements from our numpy_array:

1
2
3
4
5
6
# Accessing the first element
first_element = numpy_array[0]
second_element = numpy_array[1]
 
# Slicing elements
sub_array = numpy_array[1:3]  # Outputs np.array([2, 3])

Step 5: Using Multidimensional Arrays

Both Python lists and NumPy arrays can be multidimensional. For example, you can create a 2D array (matrix) using nested lists or NumPy arrays:

1
2
3
4
5
6
7
# Using nested lists
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
first_row = nested_list[0]
 
# Using NumPy
numpy_matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
first_row_numpy = numpy_matrix[0, :]

Full Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import numpy as np
 
# Lists
integer_list = [1, 2, 3, 4, 5]
first_element = integer_list[0]
 
# NumPy Arrays
numpy_array = np.array([1, 2, 3, 4, 5])
first_element_numpy = numpy_array[0]
 
# Multidimensional Arrays
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
first_row = nested_list[0]
 
numpy_matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
first_row_numpy = numpy_matrix[0, :]

Conclusion

In conclusion, Python lists and NumPy arrays are both excellent tools for storing and managing data.

While lists are more general-purpose, NumPy arrays are designed for numerical operations and provide a much better performance, especially for large datasets or complex operations.

Depending on your needs, you can select between using Python lists or NumPy arrays for storing and manipulating your data.