Solucionat: dividir numpy en trossos d'igual mida

Darrera actualització: 09/11/2023

Numpy és una biblioteca potent per a la computació numèrica en Python. Una tasca habitual en la computació numèrica i l'anàlisi de dades és dividir una matriu en trossos de la mateixa mida. En aquest article s'explorarà com aconseguir-ho amb Numpy i proporcionarà una guia completa sobre els passos implicats. Submergem-nos!

Per resoldre el problema de dividir una matriu Numpy gran en fragments més petits de la mateixa mida, podem utilitzar la funció numpy.split . Aquesta funció ens permet dividir una matriu en múltiples submatrius que tenen la mateixa mida al llarg d'un eix especificat. Anem a aprofundir en la solució i entendre el codi pas a pas.

import numpy as np

def numpy_split_to_chunks(array, chunk_size):
    return np.array_split(array, chunk_size, axis=0)

large_array = np.random.randint(0, 100, size=(10, 4))
chunk_size = 2
chunks = numpy_split_to_chunks(large_array, chunk_size)

Primer, importem la biblioteca numpy i després definim una funció anomenada numpy_split_to_chunks que accepta dos paràmetres d'entrada: la matriu numpy que cal dividir i la mida desitjada del fragment. La funció retorna una llista de matrius numpy, que són els fragments.

Aquí, utilitzem la funció numpy array_split per dividir la matriu d'entrada. També especifiquem l'eix al llarg del qual volem dividir la matriu. En el nostre exemple, establim axis=0, cosa que significa que volem dividir la matriu al llarg de les files.

Finalment, creem una matriu numpy aleatòria d'enters (large_array) i definim una mida de fragment (en aquest cas, 2). Cridem la nostra funció numpy_split_to_chunks per obtenir la llista de fragments.

Biblioteca Numpy

  • La biblioteca Numpy és la biblioteca bàsica per a la informàtica científica a Python.
  • S'utilitza àmpliament per a tasques relacionades amb l'àlgebra lineal, l'estadística i l'anàlisi de dades.
  • Proporciona un objecte de matriu multidimensional d'alt rendiment i eines per treballar amb matrius.

La biblioteca Numpy té una àmplia gamma de característiques i funcions que són útils per a diversos propòsits matemàtics i computacionals. Les seves capacitats inclouen la manipulació de matrius , operacions matemàtiques sobre matrius i funcions estadístiques . Numpy sovint es combina amb altres biblioteques com Matplotlib per a la visualització de dades, cosa que la converteix en un element bàsic per als científics i enginyers de dades que treballen amb Python.

Divisió de matriu Numpy

  • Numpy té diverses funcions per dividir matrius, com ara numpy.dividir, numpy.array_split, numpy.hsplit i numpy.vsplit.
  • Aquestes funcions ens permeten dividir una matriu en diverses parts al llarg d'un eix especificat.
  • Són útils per distribuir dades, càlculs paral·lels i organitzar dades.

Les diferents funcions de divisió de matrius proporcionades per Numpy permeten als desenvolupadors treballar de manera eficient amb grans conjunts de dades, extreure porcions específiques de dades o dividir dades en múltiples tasques paral·leles. Aquestes funcions són eines potents en tasques d'anàlisi i manipulació de dades, i són essencials en els fluxos de treball de programació per a la computació numèrica a Python.

Articles Relacionats: