Sentry Answers>Python>

Define a two-dimensional array in Python

Define a two-dimensional array in Python

David Y.

—

The Problem

How can I create an empty two-dimensional array in Python, e.g. for matrix operations?

The Solution

The best way to create two-dimensional (2D) arrays or matrices is by using Python’s numpy module, which provides comprehensive and performant functionality for manipulating multidimensional arrays. We can create a 2D array by passing a tuple to the zeros function. The following code will create a 3-by-3 2D array with all values set to zero:

Click to Copy
import numpy

matrix = numpy.zeros((3,3))

# array([[0., 0., 0.],
#        [0., 0., 0.],
#        [0., 0., 0.]])

Note that numpy.matrix is deprecated and should not be used for this operation.

To create a 2D array without using numpy, we can initialize a list of lists using a list comprehension. This will be slower and deprive us of the use of numpy’s matrix operations, but may be desirable in cases where we don’t want to add the dependency. We can use code similar to the following:

Click to Copy
matrix = [[0 for x in range(3)] for y in range(3)] # x = width, y = height
# [[0, 0, 0], [0, 0, 0], [0, 0, 0]]

In both of these structures, we can read and write individual elements with the following syntax:

Click to Copy
matrix[0][0] = 1 # set top-left corner to 1

print(matrix[2][2]) # print value of bottom-right corner
  • Sentry BlogPython Performance Testing: A Comprehensive Guide
  • Syntax.fmListen to the Syntax Podcast
  • Sentry BlogLogging in Python: A Developer’s Guide
  • Syntax.fm logo
    Listen to the Syntax Podcast

    Tasty treats for web developers brought to you by Sentry. Get tips and tricks from Wes Bos and Scott Tolinski.

    SEE EPISODES

Considered “not bad” by 4 million developers and more than 100,000 organizations worldwide, Sentry provides code-level observability to many of the world’s best-known companies like Disney, Peloton, Cloudflare, Eventbrite, Slack, Supercell, and Rockstar Games. Each month we process billions of exceptions from the most popular products on the internet.

learn more about sentryjoin our discord
© 2024 • Sentry is a registered Trademark of Functional Software, Inc.