我有两个列表。而且我需要基于另一个列表列表(位置)的长度来拆分numpy.ndarray(矩阵)。例如:
position: [[0.0056, 0.0065, 0.008], [0.009, 0.1, 0.127], [0.232, 0.879]]
在此示例中,矩阵的大小为(5,8)
matrix: array([[0, 0, 1, 0, 0, 1, 1, 1],
[1, 1, 0, 1, 1, 1, 1, 1],
[0, 1, 0, 1, 0, 1, 0, 1],
[1, 1, 1, 0, 0, 0, 0, 0],
[1, 1, 0, 0, 1, 0, 1, 0]])
因此,例如,位置的第一个索引中的列表的长度为3。因此,我想生成一个在位置3处削减矩阵的输出。因此,对于第一个列表的输出将是:
0, 0, 1
1, 1, 0
0, 1, 0
1, 1, 1
1, 1, 0
类似地,第二个位置索引中的列表长度也是3,因此,它将从矩阵中提取下三列,输出将是:
0, 0, 1
1, 1, 1
1, 0, 1
0, 0, 0
0, 1, 0
最后一个位置索引具有长度为2的列表,因此它将采用矩阵的最后两列。
我不知道该如何实现。帮助将不胜感激。