Comment convertir un tableau NumPy en liste en Python

Introduction

Avec NumPy, les objets np.array peuvent être convertis en liste avec la fonction tolist(). La fonction tolist() n'accepte aucun argument. Si le tableau est unidimensionnel, une liste avec les éléments du tableau est renvoyée. Pour un tableau multidimensionnel, une liste imbriquée est renvoyée.

Conditions préalables

Pour compléter ce tutoriel, vous aurez besoin de :

• Familiarité avec l'installation de Python 3 et l'utilisation de pip pour installer des packages. Et familiarité avec la programmation en Python. Comment coder en Python 3 série ou utiliser VS Code pour Python.

Ce tutoriel a été testé avec Python 3.9.6 et NumPy 1.23.3.

Conversion d'un tableau NumPy unidimensionnel en liste

Construisons un tableau unidimensionnel de [1, 2, 3]:

import numpy as np

1d array to list
arr_1 = np.array([1, 2, 3])

print(f'NumPy Array:\n{arr_1}')

Ce code donnera :

NumPy Array:
[1 2 3]

Maintenant, utilisons tolist() :

import numpy as np

1d array to list
arr_1 = np.array([1, 2, 3])

print(f'NumPy Array:\n{arr_1}')

list_1 = arr_1.tolist()

print(f'List: {list_1}')

Ce nouveau code va produire :

List: [1, 2, 3]

Le tableau a été converti de scalaires numpy en scalaires Python.

Conversion d'un tableau NumPy multidimensionnel en liste

Construisons un tableau multidimensionnel de [ [1, 2, 3], [4, 5, 6] ] :

import numpy as np

2d array to list
arr_2 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'NumPy Array:\n{arr_2}')

Ce code donnera :

NumPy Array:
[[1 2 3]
 [4 5 6]]

Maintenant, utilisons tolist() :

import numpy as np

2d array to list
arr_2 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'NumPy Array:\n{arr_2}')

list_2 = arr_2.tolist()

print(f'List: {list_2}')

Ce nouveau code va produire :

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

Le tableau a été converti de scalaires numpy en scalaires Python.

Conclusion

Dans cet article, vous avez appris à utiliser tolist() pour convertir des objets np.array en listes. Cela s'applique aux tableaux unidimensionnels et multidimensionnels.

Références

• Documentation de l'API