在 Python 中展平一个浅列表

这是一个在Python中展平一个浅列表的例子。

首先,我们定义一个函数`flatten`,它接受一个列表作为参数。在这个函数中,我们将使用列表推导式来遍历输入列表中的每个元素。如果当前元素是列表,我们就递归地调用`flatten`函数;如果不是列表,我们就将其添加到结果列表中。

```python
def flatten(lst):
    return [item for sublist in lst for item in (sublist if isinstance(sublist, list) else [sublist])]
```

然后,我们定义一个测试用例来验证我们的函数是否正确工作。

```python
def test_flatten():
    assert flatten([1, 2, [3, 4], [5, [6, 7]]]) == [1, 2, 3, 4, 5, 6, 7]
    assert flatten([[1, 2, [3, 4]], 5]) == [1, 2, 3, 4, 5]
    print("All test cases passed.")

test_flatten()
```

最后,我们使用这个函数来展平一个列表。

```python
flat_list = flatten([1, 2, [3, 4], [5, [6, 7]]])
print(flat_list)  # Output: [1, 2, 3, 4, 5, 6, 7]
```

这个函数可以通过递归的方式来展平列表,如果遇到嵌套的列表,就继续展开。这种方法简单且易于理解,但是需要注意的是,对于非常大的列表或者深层次嵌套的列表,这种方式可能会导致栈溢出。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

潮易

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值