web-dev-qa-db-fra.com

taille du tableau NumPy

Existe-t-il un équivalent du MATLAB?

 size()

commande dans Numpy?

Dans MATLAB,

>>> a = zeros(2,5)
 0 0 0 0 0
 0 0 0 0 0
>>> size(a)
 2 5

En python,

>>> a = zeros((2,5))
>>> 
array([[ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.]])

>>> ?????
28
abalter

Ceci s'appelle la "forme" dans NumPy, et peut être demandé via l'attribut .shape:

>>> a = zeros((2, 5))
>>> a.shape
(2, 5)

Si vous préférez une fonction, vous pouvez également utiliser numpy.shape(a).

54
Sven Marnach

Oui, numpy a une fonction de taille, forme et taille ne sont pas tout à fait les mêmes.

Contribution

import numpy as np
data = [[1, 2, 3, 4], [5, 6, 7, 8]]
arrData = np.array(data)

print(data)
print(arrData.size)
print(arrData.shape)

Sortie

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

8 # taille

(2, 4) # forme

5
tcratius