Python-如何基于索引数组从数组中提取元素?

Ale*_*exT 0 python arrays extraction

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

X = [1, 2, 3, 4, 5, 6, 7]
Y = [0, 3, 4]
Run Code Online (Sandbox Code Playgroud)

Python中是否有一个函数可以允许X根据提供的索引从中提取元素Y?执行后,X将是:

X = [1, 4, 5]
Run Code Online (Sandbox Code Playgroud)

Pru*_*une 5

X = [X[index] for index in Y]
Run Code Online (Sandbox Code Playgroud)

这是一个清单理解 ; 您可以查询该主题以了解更多信息。