Python 2D List ExamplesCreate a list of lists, or a 2D list. Append empty lists to a list and add elements.
dot net perls
2D list. Programs often must model 2 dimensions, not just one. In a Python list we can place other lists: this creates a 2D representation of objects.
Python list notes.
We can access elements on each list with an index—we must specify the X and Y positions. To construct a 2D list, we can use append() or an initializer.
We create an empty list and add empty lists to it with append(). We can construct any rectangular (or jagged) list this way. In this example we build a 2 by 2 list.
Step 1 We first create an empty list with the empty square brackets. Then we call the append method and add 2 empty lists.
Step 2 We access each sub-list and call append on it. This appends to the inner lists.
Step 3 We display the element at indexes 0, 0 and this value is 1. With the print() method we can display the entire list contents.
Step 4 This loop can iterate rows and columns in the 2D list. Nested for-loops loop over rows and columns.
# Step 1: create a list.
# ... Append empty lists in first two indexes.
Python program that creates and adds to 2D list
elements = 
# Step 2: add elements to empty lists.
# Step 3: display top-left element.
# ... Display entire list.
# Step 4: loop over rows.
row in elements:
# Loop over columns.
column in row:
[[1, 2], [3, 4]]
The term "jagged" implies that sub lists have uneven lengths. Here we create a list of 2 lists—one of length 2, the other of length 5. We display their lengths.
Info This is a list of lists, not exactly a 2D list. It does not need to be rectangular—some lists can be longer than others.
Tip All the methods on list, including "len" and "in," work on nested lists. But they consider only one sub-list at a time.
Caution For performance, nested lists are not optimal. An array is faster than a list.
# A jagged list.
Python program that uses jagged lists
values = [[10, 20], [30, 40, 50, 60, 70]]
for value in values:
# Print each row's length and its elements.
2 [10, 20]
5 [30, 40, 50, 60, 70]
It is possible to index into a 1D array as though it is two-dimensional. We add the first coordinate, "x," and then multiply the second "y" coordinate by the length.
Index The multiplication of the coordinates returns a single integer for a 2D point.
Here We define get_element and set_element methods. We compute indexes based on an "x" and "y" coordinate pair.
Python program that uses flattened list
get_element(elements, x, y):
return elements[x + (y * 4)
set_element(elements, x, y, value):
elements[x + (y * 4)
] = value
# Create a list of 16 elements.
elements = 
i in range(0, 16):
i = 0
x in range(0, 4):
y in range(0, 4):
# Set each element in the flattened list.
set_element(elements, x, y, i)
i += 1
# Print some elements.
print(get_element(elements, 0, 0))
print(get_element(elements, 1, 1))
print(get_element(elements, 3, 3))
Benchmark, nested list.
Suppose it is simple to change a nested list to a flattened, 1D list. This will likely result in a performance increase.
Version 1 This version of the code accesses an element in a nested list—so it accesses 2 elements total.
Version 2 This code accesses the flattened list, using an expression to compute the correct index.
Result Accessing the element at position coordinates 1, 2 is slightly faster in the flattened list.
Python program that times nested, flattened list
# Nested, 3x2.
nested_list = 
nested_list.append([10, 20, 30])
nested_list.append([40, 50, 60
# Flattened, 3x2.
flat_list = [10, 20, 30, 40, 50, 60
# Test the two lists.
print(flat_list[1 + (2 * 2)])
# Version 1: access element in nested list.
i in range(0, 100000000):
 != 60:
# Version 2: access element in flattened list.
i in range(0, 100000000):
[1 + (2 * 2)] != 60:
1420077375.334: Nested list, 0.539 s
1420077375.794: Flat list, 0.460 s
Python supports a special "array" from the array module. An integer array is more compact in memory than an integer list. We can create a flattened 2D array.
A review. Sometimes a 2D list is helpful in programs. For a simple or small coordinate system or grid, nested lists can be useful. They are flexible. The syntax is clear.
Larger data sizes. For larger or more complex requirements, this approach is not ideal. Another data structure, like an array, is a better choice at this scale.
© 2007-2021 sam allen. send bug reports to firstname.lastname@example.org.