我目前正在用python编程,我创建了一个从用户输入列表的方法,不知道他是多维的还是一维的。我该如何检查?样本:
def __init__(self,target):
for i in range(len(target[0])):
w[i]=np.random.rand(len(example[0])+1)
目标是列表。问题是目标 [0] 可能是 int。
我目前正在用python编程,我创建了一个从用户输入列表的方法,不知道他是多维的还是一维的。我该如何检查?样本:
def __init__(self,target):
for i in range(len(target[0])):
w[i]=np.random.rand(len(example[0])+1)
目标是列表。问题是目标 [0] 可能是 int。
我想你只想要isinstance?
示例用法:
>>> a = [1, 2, 3, 4]
>>> isinstance(a, list)
True
>>> isinstance(a[0], list)
False
>>> isinstance(a[0], int)
True
>>> b = [[1,2,3], [4, 5, 6], [7, 8, 9]]
>>> isinstance(b[0], list)
True
根据评论,无论如何您都将输入转换为 numpy 数组。由于np.array
已经处理了计算输入列表嵌套的深度,因此从该数组中找出维数比从嵌套列表中更容易。
特别是,数组有一个shape
属性,它是数组沿每个维度的长度的元组,所以len(myarray.shape)
会告诉你维度的数量。例如,
>>> import numpy as np
>>> a = np.array([[1,2,3],[1,2,3]])
>>> len(a.shape)
2
如果您想了解列表有多少维度,可以使用以下代码片段:
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)
这是非常简化的解决方案。在大多数情况下,多维列表/数组/矩阵将在第一个索引中包含一个列表对象。话虽如此,在 python 中,因为您不需要定义数据类型,如果您的列表看起来像链接:[1,[2,3],4],这可能会返回不正确。-- 否则它应该适用于所有正常的多维列表。
def is2DList(matrix_list):
if isinstance(matrix_list[0], list):
return True
else:
return False
# list
list_1 = [1,2,3,4] # 1 x 4
list_2 = [ [1,2,3],[3,4,5] ] # 2 x 2
list_3 = [1, [2,3], 4]
print(is2DList(list_1)) # False
print(is2DList(list_2)) # True
print(is2DList(list_3)) # False - incorrect result