我們索引列表的第 i 個元素表示我們想知道的 2D 列表中第 i 個 1D 列表的索引。如果我們的二維列表和索引列表如下所示:a = [[1,7,3],[9,0,8]]list_ix = [1, 0]期望的結果是 [7,9]。目前我通過使用如下循環來做到這一點:res = []for i in range(len(list_ix)): res.append(a[i][list_ix[i]])但這在實際代碼中看起來很亂。有一個更好的方法嗎?我將不勝感激任何建議。謝謝你!
如何根據索引列表從二維列表中的每個一維列表中選擇一個元素?
寶慕林4294392
2023-06-20 16:39:15