Tamanho do conjunto NumPy

Existe um equivalente ao MATLAB 

 size()

comando em Numpy?

em MATLAB, 

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

Em Python, 

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

>>> ?????
Author: Amro, 2012-06-20
Source

2 answers

Isto é chamado de "forma" em NumPy, e pode ser solicitado através do atributo .shape: 

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

Se preferir uma função, também pode utilizar numpy.shape(a).

 49
Author: Sven Marnach, 2012-06-20 18:02:01

Sim numpy tem uma função de tamanho, e a forma e o tamanho não são exactamente os mesmos.

Entrada 

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)

Saída

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

8 # Tamanho

(2, 4) # Forma

 0
Author: Conrad Thiele, 2018-10-02 13:52:48