python判断列表维度_python：检查列表是多维还是一维

如果您想知道一个列表有多少维度，可以使用以下代码片段：def test_dim(testlist, dim=0):

"""tests if testlist is a list and how many dimensions it has

returns -1 if it is no list at all, 0 if list is empty

and otherwise the dimensions of it"""

if isinstance(testlist, list):

if testlist == []:

return dim

dim = dim + 1

dim = test_dim(testlist[0], dim)

return dim

else:

if dim == 0:

return -1

else:

return dim

a=[]

print test_dim(a)

a=""

test_dim(a)

print test_dim(a)

a=["A"]

print test_dim(a)

a=["A", "B", "C"]

print test_dim(a)

a=[[1,2,3],[1,2,3]]

print test_dim(a)

a=[[[1,2,3],[4,5,6]], [[1,2,3],[4,5,6]], [[1,2,3],[4,5,6]]]

print test_dim(a)