Numpy中的数组索引缩写

Question

我有一个函数，以一个数组作为输入。数组的维度在1和4之间。我想检查第一个维度的长度是否为2，如果长度大于2，则将其减为2。我正在做的是。

if D.shape[0] != 2: 
   if D.ndim == 1:
       D = D[:2]
   elif D.ndim == 2:
       D = D[:2, :]
...

正如你所看到的，这很糟糕。所以我想知道如何对类似D = D[:2,......的东西进行检查。

Answer 1

D=D[:2] ...对于所有的情况来说应该是足够的。默认情况下是取所有内维对象（就像列表，就像其他语言中的数组）。