以下是一个示例代码来按照另一个列表从数组中提取元素:
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]。