如何找到嵌套元组的维度？

Question

我有一个看起来像这样的元组

  array=(1,2,3,4)
  lenM = numpy.shape(array)
  print lenM
  (4,)

  if not lenM[1]:
       "Code"

现在我如何自动化我的代码来查找元组是一维还是二维？

Answer 1

你可以使用numpy.ndim这个:

In [4]: np.ndim((1,2,3,4))
Out[4]: 1
In [5]: np.ndim(((1,2),(3,4)))
Out[5]: 2

Answer 2

array=(1,2,3,4)
lenM = numpy.shape(array)
print lenM
(4,)

if len(lenM) == 1:
    "1 dimensional code"
elif len(lenM) == 2:
    "2 dimensional code"

len(lenM)会告诉你数组中是否有多个维度.如果len(lenM)为1,则只有一个维度.如果数组有多个维度,lenM将有多个元素.