Cómo calcular la diferencia entre elementos vecinos en una array usando NumPy

Veamos cómo calcular la diferencia entre elementos vecinos en una array usando la biblioteca NumPy.

Entonces, podemos encontrar la diferencia entre los elementos vecinos usando la función numpy.diff() de la biblioteca NumPy.

Sintaxis: numpy.diff(arr, n, eje)

Ahora, veamos un ejemplo:

Ejemplo 1:

Python3

# import library
import numpy as np
  
# create a numpy 1d-array
arr = np.array([1, 12, 3, 14, 5,
              16, 7, 18, 9, 110])
  
# finding the difference between
# neighboring elements
result = np.diff(arr)
  
print(result)

Producción:

[ 11  -9  11  -9  11  -9  11  -9 101]

Ejemplo 2:

Python3

# import library
import numpy as np
  
# create a numpy 2d-array
arr = np.array([[10, 12, 14], 
                [25, 35, 45],
                [12, 18, 20]])
  
# finding the difference between
# neighboring elements along row
result = np.diff(arr, axis = 1)
  
print(result)

Producción:

[[ 2  2]
[10 10]
[ 6  2]]

Ejemplo 3:

Python3

# import library
import numpy as np
  
# create a numpy 2d-array
arr = np.array([[10, 12, 14], 
                [25, 35, 45],
                [12, 18, 20]])
  
# finding the difference between
# neighboring elements along column
result = np.diff(arr, axis = 0)
  
print(result)

Producción:

[[ 15  23  31]
[-13 -17 -25]]

Publicación traducida automáticamente

Artículo escrito por vasu_gupta y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *