我想将二进制 numpy.ndarray 转换为列表列表 python？

Question

您好，我需要将二进制“numpy.ndarray”转换为列表列表，如下例所示：

Matrix:
[[  #   0  1  2  3  4]
 [  0   1  0  1  0  1]
 [  1   0  0  1  1  1]
 [  2   1  0  1  1  0]
 [  3   0  0  1  0  1]
 [  4   1  1  0  0  1]
 [  5   1  0  1  1  1]
 [  6   0  0  1  0  1]]

List:
[[0,2,4],[2,3,4],[0,2,3],[2,4],[0,1,4],[0,2,3,4],[2,4]]

提前谢谢

Answer 1

这是一个显示解决方案的小例子

import numpy as np

x = np.array([[  0 , 1,  0 , 1 , 0 , 1],
              [  1 ,  0 , 0,  1 , 1,  1],
              [  2  , 1 , 0  ,1 , 1,  0],
              [  3 ,  0 , 0  ,1,  0 , 1],
              [  4 ,  1,  1 , 0 , 0 , 1],
              [  5 ,  1  ,0 , 1,  1  ,1],
              [  6 ,  0 , 0,  1 , 0 , 1]])

out = []
for row in x:
  out.extend([np.where(row[1:]==1)[0].tolist()])

print(out)