我想以一種特殊的方式合并python中的兩個數組。我的輸出數組的奇數索引的out條目應該是我的第一個輸入數組的對應條目in0。具有偶數索引的條目out應是我的第二個輸入數組的對應條目 in1。in0,in1并且out都是相同的長度。例子：輸入數組in0 = [0, 1, 2, 3]
in1 = [4, 5, 6, 7]應合并到輸出數組out = [0, 5, 2, 7]有沒有比循環輸入的整個長度并out“手動”填充我的更好的方法？
3 回答
互換的青春
TA貢獻1797條經驗 獲得超6個贊
您可以使用列表理解并從in0偶數索引和in1奇數索引中選擇值：
[in0[i] if i % 2 == 0 else in1[i] for i in range(len(in0))] # [0, 5, 2, 7]
一只甜甜圈
TA貢獻1836條經驗 獲得超5個贊
如果您樂于制作完整的列表副本，那么切片很簡單：
>>> in0 = [0, 1, 2, 3]
>>> in1 = [4, 5, 6, 7]
>>> out = in0[:]
>>> out[1::2] = in1[1::2]
>>> out
[0, 5, 2, 7]
蕭十郎
TA貢獻1815條經驗 獲得超13個贊
如果你不介意一些冗長...
from itertools import cycle
in0 = [0, 1, 2, 3]
in1 = [4, 5, 6, 7]
out = [pair[i] for pair, i in zip(zip(in0, in1), cycle([0,1]))]
這個怎么運作：
zip(in0, in1)是一個元組序列，(0,4), (1,5), (2,6), (3,7)。cycle([0,1])是無盡的交替 0 和 1 流，用作步驟 1 中元組中的索引。zip(zip(...), cycle(...))產生一對元組和索引：(0, (0,4)), (1, (1,5)), (0, (2,6)), (1, (3,7)).
列表理解從每個元組中獲取正確的元素。
最后，列表推導式是一個通用版本
[(0,4)[0], (1,5)[1], (2,6)[0], (3,7)[1]]
添加回答
舉報
0/150
提交
取消