如何通过Python中的行列表获取切片2D数组

Question

我有一个2D数组，我需要使用选定的行获得另一个2D数组。

这是针对推荐引擎的，在该引擎中我执行了奇异值分解，现在我只需要从U矩阵中获取一些特定的行。我有这些行的索引，但是我不知道如何根据索引行对矩阵进行“重塑”。

为简化起见，我在下面用一个基本示例说明了我想做什么。

IN

a=np.diag(np.arange(14))

rows=np.array([0,3,5])

OUT

#expected output

[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
 [0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
 [0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 0]]

有支持吗？