我有一个看起来像这样的元组
array=(1,2,3,4) lenM = numpy.shape(array) print lenM (4,) if not lenM[1]: "Code"
现在我如何自动化我的代码来查找元组是一维还是二维?
你可以使用numpy.ndim
这个:
In [4]: np.ndim((1,2,3,4)) Out[4]: 1 In [5]: np.ndim(((1,2),(3,4))) Out[5]: 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将有多个元素.