numpy配列の寸法を確認する方法は？

Question

私は関数内でこのような何かをしたい：

if image.shape == 2 dimensions 
    return image # this image is grayscale 
else if image.shape = 3 dimensions 
    return image # image is either RGB or YCbCr colorspace 

ここで、画像をnumpyの配列です。私はその検査条件を定義することができません。私は本当にこの時点で立ち往生しています。誰でも助けてくれますか？

Answer 1

numpy.array.shapeは配列次元のタプルです。タプルの長さを計算することができ、それは次元の数を与えます。

if len(image.shape) == 2: 
    return image # this image is grayscale 
elif len(image.shape) == 3: 
    return image # image is either RGB or YCbCr colorspace 

ナンシーアレイには、ndim属性もあります。

if image.ndim == 2: 
    return image # this image is grayscale 
elif image.ndim == 3: 
    return image # image is either RGB or YCbCr colorspace