透過索引存取多個清單元素
許多程式設計場景需要根據索引存取清單中的特定元素。例如,您可能需要從給定清單中提取索引為 1、2 和 5 的元素。
一個常見的方法是您使用的方法:
a = [-2,1,5,3,8,5,6] b = [1,2,5] c = [ a[i] for i in b]
但是,有可以簡化此過程的替代方法。
選項 1:使用 itemgetter 運算子
from operator import itemgetter a = [-2, 1, 5, 3, 8, 5, 6] b = [1, 2, 5] itemgetter(*b)(a) # (1, 5, 5)
itemgetter 運算子可讓您有效率地執行索引查找。
選項 2:使用 Numpy
import numpy as np a = np.array([-2, 1, 5, 3, 8, 5, 6]) b = [1, 2, 5] list(a[b]) # [1, 5, 5]
Numpy 提供了一種基於索引清單對陣列進行索引的便利方法。
雖然提供的解決方案很簡潔和有效的方法,這些替代方法在某些情況下提供了潛在的性能增強或額外的靈活性。最終,選擇使用哪種方法取決於您應用程式的特定要求。
以上是如何透過索引高效存取多個清單元素?的詳細內容。更多資訊請關注PHP中文網其他相關文章!