gpt4 book ai didi

Python 二维列表切片

转载 作者:太空狗 更新时间:2023-10-30 01:21:08 28 4
gpt4 key购买 nike

我环顾四周,找不到这么具体的东西,所以这里是:

我有一个列表列表:

S = [[3,4],[6,7],[10,12]]

我想将第 i 个索引元素的第 0 个索引添加到另一个列表的末尾:

R = [5,6,7]

通常对于一维列表我可以说:

R = R + S[i:]

并获取第 i 个索引中的所有元素,但我想要 2D S 的第 i 个索引的第 0 个索引。如果我们从 i = 1 开始,我最终会得到:

R = [5,6,7,6,10]

此外,我不想使用 for 循环,我想要一个可以工作的列表切片方法(如果有的话),因为它需要在某个范围内。

最佳答案

您可以使用zip 转置矩阵:

>>> S
[[3, 4], [6, 7], [10, 12]]
>>> zip(*S)
[(3, 6, 10), (4, 7, 12)]

然后切片转置:

>>> j=0
>>> i=1
>>> zip(*S)[j][i:]
(6, 10)

元组是可迭代的,因此串联可以与列表一起使用:

>>> R = [5,6,7]
>>> R+=zip(*S)[j][i:]
>>> R
[5, 6, 7, 6, 10]

关于Python 二维列表切片,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33714135/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com