以下是一个示例代码来按照另一个列表从数组中提取元素:
def extract_elements(arr, idx_list):
extracted_elements = []
for idx in idx_list:
if idx < len(arr):
extracted_elements.append(arr[idx])
return extracted_elements
# 示例用法
arr = [1, 2, 3, 4, 5]
idx_list = [0, 2, 4]
extracted_elements = extract_elements(arr, idx_list)
print(extracted_elements) # 输出 [1, 3, 5]
在这个示例中,extract_elements
函数接受一个数组arr
和一个索引列表idx_list
作为输入。它会遍历idx_list
中的每个索引,检查索引是否在数组arr
的范围内,然后将对应的元素添加到extracted_elements
列表中。最后,函数返回extracted_elements
列表,其中包含了从数组中提取的元素。
在示例中,数组arr
是[1, 2, 3, 4, 5]
,索引列表idx_list
是[0, 2, 4]
。函数将提取索引为0、2和4的元素,即数组中的第一个、第三个和第五个元素,最后返回[1, 3, 5]
。