如何从基于索引数组的数组中提取元素？ [英] Python - How to extract elements from an array based on an array of indices?

问题描述

假设我有一个元素列表X和一个索引Y。

X = [1, 2, 3, 4, 5, 6, 7]
Y = [0, 3, 4]

在Python中是否有一个函数允许用户根据Y中提供的索引从X中提取元素？执行后，X为：

X = [1, 4, 5]

推荐答案

X = [X[index] for index in Y]

这是列表理解；您可以查看该主题以了解更多信息。

这篇关于如何从基于索引数组的数组中提取元素？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！