HOWTO · Python

Declare 3D Array em Python

Existem 3 métodos principais que podem ser usados para declarar um array 3D em Python, as compreensões de lista, o método de multiplicação e o pacote numpy.

Neste tutorial, discutiremos métodos para declarar arrays tridimensionais em Python.

Declarar lista 3D usando abrangências de lista em Python

No que diz respeito à funcionalidade básica, as listas fazem o mesmo trabalho que os arrays em Python. A compreensão de lista é uma forma de realizar operações complexas em listas. As compreensões de lista também podem ser usadas para declarar un array 3D. O exemplo de código a seguir nos mostra como podemos usar as compreensões de lista para declarar um array 3D em Python.

n = 3

distance = [[[0 for k in range(n)] for j in range(n)] for i in range(n)]

print(distance)

Resultado:

[[[0, 0, 0], [0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0], [0, 0, 0]]]

No código acima, primeiro declaramos o número de dimensões e, em seguida, inicializamos uma lista de n dimensões usando compreensões de lista.

Declare a lista 3D usando o método de multiplicação em Python

O método de compreensão de lista funciona bem, mas é um código um pouco extenso. Se quisermos minimizar nosso código, podemos usar outra abordagem chamada método de multiplicação. O exemplo de código a seguir nos mostra como usar o método de multiplicação para declarar un array 3D em Python.

distance = [[[0] * n] * n] * n

print(distance)

Resultado:

[[[0, 0, 0], [0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0], [0, 0, 0]]]

O código acima faz a mesma coisa que as compreensões de lista, mas reduz o código drasticamente.

Declare 3D Array usando o pacote NumPy em Python

Se quisermos realizar algumas operações especificamente em arrays em Python, é melhor usar o pacote NumPy. É um pacote projetado especificamente para trabalhar com arrays em Python.

NumPy é um pacote externo e não vem pré-instalado com Python. Precisamos instalá-lo antes de usá-lo. O comando para instalar o pacote NumPy é fornecido abaixo.

pip install numpy

O exemplo de código a seguir mostra como podemos declarar um array tridimensional em Python usando o pacote NumPy.

import numpy as np

i = 3
j = 3
k = 3

new_array = np.zeros((i, j, k))

print(new_array)

Resultado:

[[[0. 0. 0.]
  [0. 0. 0.]
  [0. 0. 0.]]

 [[0. 0. 0.]
  [0. 0. 0.]
  [0. 0. 0.]]

 [[0. 0. 0.]
  [0. 0. 0.]
  [0. 0. 0.]]]

No código acima, primeiro declaramos o número de elementos em cada dimensão do array com i, j e k. Depois disso, passamos essas dimensões para np.zeros() para inicializar um array 3D. O np.zeros() método nos dá un array e preenche todos os elementos com um 0.