用python实现看list的形状
时间: 2023-10-07 15:14:29 浏览: 147
圣诞树源码(python)
以下是检查列表形状的Python代码:
```python
def check_shape(lst):
shape = []
while isinstance(lst, list):
shape.append(len(lst))
lst = lst[0]
return tuple(shape)
# 示例
lst = [[1, 2], [3, 4], [5, 6]]
print(check_shape(lst)) # 输出 (3, 2)
```
这个函数接受一个嵌套列表作为参数,并返回一个元组,其中每个元素表示列表在相应维度上的大小。例如,如果列表是 `[[1, 2], [3, 4], [5, 6]]`,则形状为 `(3, 2)`,表示它有三行和两列。
阅读全文