python中如何得到未知list的维度
时间: 2024-02-21 13:00:54 浏览: 83
Python3 Tensorlfow:增加或者减小矩阵维度的实现
在 Python 中,可以使用内置的 `len()` 函数来获取列表的长度,即列表中元素的个数,从而确定列表的维度。如果列表是多维的,则可以递归地使用 `len()` 函数来获取每个维度上的长度。
以下是一个示例代码,可以获取任意维度的列表的维度信息:
```python
def get_shape(lst):
shape = []
while isinstance(lst, list):
shape.append(len(lst))
lst = lst[0]
return tuple(shape)
# 示例
lst = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]
print(get_shape(lst)) # 输出 (2, 2, 2)
```
在上述示例中,`get_shape()` 函数接受一个列表作为参数,返回一个元组,其中包含列表的每个维度上的长度。该函数使用了递归的方式,每次处理一个维度,并将该维度的长度添加到 `shape` 列表中。最后,将 `shape` 列表转换为元组并返回。
阅读全文