根据另一个列表的长度拆分凹凸不平的数组

时间:2019-06-23 11:00:27

标签: list python-3.6 numpy-ndarray

我有两个列表。而且我需要基于另一个列表列表(位置)的长度来拆分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的列表,因此它将采用矩阵的最后两列。

我不知道该如何实现。帮助将不胜感激。

0 个答案:

没有答案