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

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