Du tableau numpy à la liste : méthode de conversion facile à comprendre

En science des données et en machine learning, l'utilisation de ndarray (tableaux multidimensionnels) de la bibliothèque numpy est essentielle. Cependant, nous devons parfois convertir ces tableaux en listes Python pour un traitement ultérieur. Cet article fournira une méthode de conversion simple et facile à comprendre d'un tableau numpy en liste, et fournira des exemples de code spécifiques.

1. Utilisez la méthode tolist()

ndarray dans numpy fournit la méthode tolist(), qui peut être convertie en liste Python. Voici un exemple simple :

import numpy as np

nd_array = np.array([[1, 2, 3], [4, 5, 6]])
lst = nd_array.tolist()

print(lst)

Sortie :

[[1, 2, 3], [4, 5, 6]]

2. Utilisez la méthode flatten() combinée avec la méthode tolist()

En plus de la méthode tolist() mentionnée précédemment, nous pouvons également convertir ndarray en unidimensionnel. array by , puis utilisez la méthode tolist() pour le convertir en liste Python. Voici un exemple :

import numpy as np

nd_array = np.array([[1, 2, 3], [4, 5, 6]])
flat_array = nd_array.flatten()
lst = flat_array.tolist()

print(lst)

Sortie :

[1, 2, 3, 4, 5, 6]

3. Utilisez la méthode list()

En plus de la méthode tolist() dans la bibliothèque numpy, Python est également livré avec une méthode list() qui convertit les objets itérables. en listes. Nous pouvons transmettre l'objet ndarray directement à la méthode list() pour la conversion. Voici un exemple :

import numpy as np

nd_array = np.array([[1, 2, 3], [4, 5, 6]])
lst = list(nd_array)

print(lst)

Sortie :

[array([1, 2, 3]), array([4, 5, 6])]

Il convient de noter que l'utilisation de la méthode list() créera une liste contenant plusieurs objets ndarray unidimensionnels.

4. Utiliser la boucle for

Enfin, nous pouvons utiliser la boucle for de Python pour convertir l'objet ndarray en une liste Python. Voici un exemple :

import numpy as np

nd_array = np.array([[1, 2, 3], [4, 5, 6]])
lst = []

for row in nd_array:
    for value in row:
        lst.append(value)

print(lst)

Sortie :

[1, 2, 3, 4, 5, 6]

Ce qui précède est une méthode de conversion simple et facile à comprendre d'un tableau numpy en liste, y compris l'utilisation de la méthode tolist(), combinant la méthode flatten() et la tolist (), en utilisant la méthode list() et utilisez une boucle for. Toutes les méthodes ci-dessus peuvent être converties efficacement, choisissez simplement la méthode qui vous convient le mieux en fonction de la situation réelle.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!