将数据放入特定形状的列表中

我有一个如下所示的列表：

my_list = [[20, 15, 10], [15, 22, 37, 46], [22, 91]]

所以它是二维的，但并非每条线都有相同数量的元素。

我现在有一个平面 ndarray，例如：

my_ndarray = np.array([9, 2, 4, 4, 1, 6, 7, 8, 17])

与 my_list 具有相同数量的元素。现在我想将 my_ndarray 调整为与 my_list 相同，即：

my_ndarray = [[9, 2, 4], [4, 1, 6, 7], [8, 17]]

所以我们可以注意到，my_list 和 my_ndarray 都包含 3 个子列表，第一个子列表包含 3 个元素，第二个子列表包含 4 个元素，第三个子列表包含 2 个元素。

有一个巧妙的方法来做到这一点吗？

Thanks!

>>> it = iter([9, 2, 4, 4, 1, 6, 7, 8, 17])
>>> my_list = [[20, 15, 10], [15, 22, 37, 46], [22, 91]]
>>> [[next(it) for i in j] for j in my_list]
[[9, 2, 4], [4, 1, 6, 7], [8, 17]]