如何从基于索引数组的数组中提取元素？

问题描述

假设我有一个元素列表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]

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