逻辑或列

Question

在 Python 中对矩阵的每一列执行逻辑 OR（在 1 和 0 上）的最有效方法是什么？

   0    0    0
   0    0    0
   1    0    1
   0    0    1
OR 0    0    1
   _    _    _
   1    0    1

一些背景：我正在使用一种动态编程方法来为我一直在从事的一个小项目生成一个表。我有两张桌子，一张是 3-d，另一张是 2-d。在 3-d 表中只有 1 或 0，而另一个包含整数。

2-d 表与 3-d 表一样生成，所以我必须将它们交错。我想要 OR 整个列的原因是查看其中是否有任何 1，如果是，则同一列将在 2-d 表中包含 1。

例子：

3-d 表：

[1][i][j]
1000010000000000...
1000000001000000...
1000000000010000...
1000000000000001...

[2][i][j]
1000100000000000...
1000001000000000...
1000000001000000...

二维表： （在对 3-d 表的每一列进行 OR 运算后，其中最左边的索引对应于 2-d 表中的一行）：

1000010001100001...
100010100000000...

不确定这是否全部清楚...但感谢任何提供帮助的人！ :D

我不会解释我使用的方法

Answer 1

您如何表示这些数据？如果这是列表列表，则交换 3D 数组的最后两个索引，因此“列”可以作为顺序列表访问，并使用 any。这与普通 CPython 中的速度差不多：循环将在 C 运行时中完全执行。

Answer 2

使用numpy：

>>> from numpy import array
>>> m = array([[0,0,0],[0,0,0],[1,0,1],[0,0,1],[0,0,1]])
>>> m
array([[0, 0, 0],
       [0, 0, 0],
       [1, 0, 1],
       [0, 0, 1],
       [0, 0, 1]])
>>> m.any(0)*1
array([1, 0, 1])
>>> len('.any(0)*1')
9

或者，不那么简洁：

>>> m.any(axis=0)
array([ True, False,  True], dtype=bool)
>>> m.any(axis=0)*1 # make them ints
array([1, 0, 1])

如果您必须使用纯 Python，我实际上会按照 @rkhayrov 的建议进行操作，如果有必要，请继续转置和索引混洗，以使引用列就像引用列表一样容易。但是写m[:, 0]和m[2:5, :]的能力太方便了，我不能轻易放弃。

Answer 3

应该这样做。 reduce 不受欢迎，map 和 filter 也是如此。

col = 2
value = 0
for row in range(height):
    value |= matrix[row][col]

对于 python 2.x，使用 xrange 代替 range。