Primeros Pasos

Instalando e importando NumPy

Para usar NumPy debemos instalar la librería desde la consola con el siguiente comando:

pip3 install numpy

Después debemos importarla en nuestro código:

import numpy as np

Usando NumPy

NumPy incluye varios objetos y funciones que podemos invocar iniciando con np., seguido del nombre del objeto o la función. Por ejemplo, para sacar el promedio de un arreglo haríamos lo siguiente:

prom = np.mean([1, 2, 3, 4])
print(prom) # 2.5

Algo importante que debemos aclarar es que, aunque en este ejemplo le pasamos una lista a la función mean, NumPy convierte la lista a un arreglo de NumPy internamente.

Podemos transformar una lista de Python en un arreglo de NumPy usando la función array:

arr = np.array([1, 2, 3])

Sobre los arreglos de NumPy podemos realizar operaciones matemáticas. Por ejemplo, podemos sumar dos arreglos elemento por elemento:

arr = np.array([1, 2]) + np.array([1, 2])
print(arr) # [2, 4]

Si intentamos hacer lo mismo sobre dos listas de Python no se suman sino que se concatenan:

[1, 2] + [1, 2] # [1, 2, 1, 2]

En la siguiente guía vamos a explorar en más detalle los arreglos de NumPy.