import numpy as np


primos = [2, 3, 7, 19, 23]

arreglo = np.array(primos)

# Transformada rápida de Fourier en un arreglo:
print(np.fft.fft(arreglo))

print('')

# Creación de una matriz:
matriz = np.array([[2, 3], [11, 23]])

# Transformada rápida de Fourier en una matriz:
print(np.fft.fft2(matriz))