python - 检查 numpy 数组是否具有正常形状

Question

如何检查 numpy 数组是否具有规则形状。

在下面的示例中，x 是一个 *2 x 3* 矩阵。然而，y 并不规则，因为它无法表示为适当的矩阵。

鉴于我有一个 numpy 数组，是否有一种方法(最好是内置的)可以用来检查 numpy 数组是否是一个实际的矩阵

In [9]: import numpy as np                                                      

In [10]: x = np.array([[1,2,3],[4,5,6]])                                        

In [11]: x.shape                                                                
Out[11]: (2, 3)

In [12]: y = np.array([[1,2,3],[4,5]])                                          

In [13]: y.shape                                                                
Out[13]: (2,)

Answer 1

两者都是数组，并且都是有效的形状。但是，对于正常，您的意思是每个元素都具有相同的形状和长度。为此，更好的方法是检查数据类型。对于可变长度的情况，它将是object。因此，我们可以检查该条件并相应地进行调用。因此，只需做 -

def is_normal_arr(a): # a is input array to be tested
    return a.dtype is not np.dtype('object')