您可以使用解析：

res1 = [[idx+1 for idx, val in enumerate(row) if val] for row in p]
res2 = [[0] if a == [] else a for a in res1] 
print(res2)

字符串
给出：

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

型

使用两级ITER来得到这个。

p = np.array([[0, 1, 0, 1],
                  [0, 0, 0, 0],
                  [1, 0, 0, 0],
                  [0, 0, 1, 0]])
    res = []
    for row in p:
        tmp = []
        for idx, value in enumerate(row):
            if value == 1:
                tmp.append(idx+1)
        if len(tmp) == 0:
            tmp.append(0)
        res.append(tmp)
    print(res)

字符串
输出量

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

型

可以使用np.nonzero

lst = [[val + 1 if val else 0 for val in np.nonzero(lst)[0]] for lst in p]
print(lst) # [[2, 4], [], [0], [3]]

字符串

使用np的简短一行程序，其中：

[list(np.where(r==1)[0]+1) or [0] for r in p]

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

字符串