gpt4 book ai didi

python - 从一维列表创建二维索引列表

转载 作者:太空宇宙 更新时间:2023-11-03 13:38:04 28 4
gpt4 key购买 nike

我有一个整数列表x,我想从中创建一个二维整数列表y。这里,y 中的每一行 i 都是 x 中具有值 i 的元素的索引列表>.

例如,如果:

x = [2, 0, 1, 1, 2, 4],

然后:

y = [[1], [2, 3], [0, 4], [], [5]]

我怎样才能在 Python 中巧妙地做到这一点?

最佳答案

或者使用列表理解:

x = [2, 0, 1, 1, 2, 4]
y = [[j for j in range(len(x)) if x[j]==i] for i in range(max(x)+1)]

关于python - 从一维列表创建二维索引列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36508027/

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