NumPy - 在 nd 数组的每一列中查找和打印非零元素

Question

假设我有以下 Numpy nd 数组：

array([[['a',0,0,0],
        [0,'b','c',0],
        ['e','d',0,0]]])

现在我想定义元素的“双重连接”，如下所示：

1. 我们将此数组中的每一列视为一个时间瞬间，并且该瞬间中的所有元素都被认为是同时发生的。 0 表示什么都没有发生。 例如，a 和 e 发生在第一个时刻，b 和 d 发生在第二个时刻，c 本身发生在第三个时刻。
2. 如果有两个元素，我相信它有“双重连接”，我想打印这样的连接（如果一列中没有这样的对，则继续到下一列直到结束）：

('a','e')
('e','a')
('b','d')
('d','b')

我试图提出迭代所有列的解决方案，但没有奏效。有人可以分享一些关于此的提示吗？

您可以通过以下命令重新创建原始数组

array = np.array([['a',0,0,0],
    [0,'b','c',0],
    ['e','d',0,0],dtype=object)

Answer 1

您可以计算每列有多少个非零元素。 您选择具有两个非零元素的列，重复它们并每隔一列反转：

pairs = np.repeat(array[(array[:, (array != 0).sum(axis=0) == 2]).nonzero()].reshape((2, -1)).T, 2, axis=0)
pairs[1::2] = pairs[1::2, ::-1]

如果你想将这些转换为你想要的输出中的元组，你可以做一个列表理解：

output = [tuple(pair) for pair in pairs]