Numpy 逐元素操作
Numpy element-wise in operation
假设我有一个长度为 n 的列向量 y,并且我有一个大小为 n*m 的矩阵 X。我想检查 y 中的每个元素 i,该元素是否在 X 中的相应行中。最有效的方法是什么?
例如:
y = [1,2,3,4].T
和
X =[[1, 2, 3],[3, 4, 5],[4, 3, 2],[2, 2, 2]]
那么输出应该是
[1, 0, 1, 0] or [True, False, True, False]
哪个更容易。
当然我们可以使用 for 循环来遍历 y 和 X,但是有没有更有效的方法呢?
使用broadcasting
-
的向量化方法
((X == y[:,None]).any(1)).astype(int)
样本运行-
In [41]: X # Input 1
Out[41]:
array([[1, 2, 3],
[3, 4, 5],
[4, 3, 2],
[2, 2, 2]])
In [42]: y # Input 2
Out[42]: array([1, 2, 3, 4])
In [43]: X == y[:,None] # Broadcasted comparison
Out[43]:
array([[ True, False, False],
[False, False, False],
[False, True, False],
[False, False, False]], dtype=bool)
In [44]: (X == y[:,None]).any(1) # Check for any match along each row
Out[44]: array([ True, False, True, False], dtype=bool)
In [45]: ((X == y[:,None]).any(1)).astype(int) # Convert to 1s and 0s
Out[45]: array([1, 0, 1, 0])
假设我有一个长度为 n 的列向量 y,并且我有一个大小为 n*m 的矩阵 X。我想检查 y 中的每个元素 i,该元素是否在 X 中的相应行中。最有效的方法是什么?
例如:
y = [1,2,3,4].T
和
X =[[1, 2, 3],[3, 4, 5],[4, 3, 2],[2, 2, 2]]
那么输出应该是
[1, 0, 1, 0] or [True, False, True, False]
哪个更容易。
当然我们可以使用 for 循环来遍历 y 和 X,但是有没有更有效的方法呢?
使用broadcasting
-
((X == y[:,None]).any(1)).astype(int)
样本运行-
In [41]: X # Input 1
Out[41]:
array([[1, 2, 3],
[3, 4, 5],
[4, 3, 2],
[2, 2, 2]])
In [42]: y # Input 2
Out[42]: array([1, 2, 3, 4])
In [43]: X == y[:,None] # Broadcasted comparison
Out[43]:
array([[ True, False, False],
[False, False, False],
[False, True, False],
[False, False, False]], dtype=bool)
In [44]: (X == y[:,None]).any(1) # Check for any match along each row
Out[44]: array([ True, False, True, False], dtype=bool)
In [45]: ((X == y[:,None]).any(1)).astype(int) # Convert to 1s and 0s
Out[45]: array([1, 0, 1, 0])