比较不同形状的numpy数组
我有两个形状的numpy数组(4,4)和(9,4)比较不同形状的numpy数组,numpy,Numpy,我有两个形状的numpy数组(4,4)和(9,4) 我需要比较matrix2的所有行和matrix1的每一行。如何在不循环matrix1元素的情况下完成此操作?如果是关于行的元素比较,请检查此示例: # Generate sample arrays a = np.random.randint(0, 5, size = (4, 3)) b = np.random.randint(-1, 6, size = (5, 3)) # Compare a == b[:, None] 最后一行为您进行比较
我需要比较matrix2的所有行和matrix1的每一行。如何在不循环matrix1元素的情况下完成此操作?如果是关于行的元素比较,请检查此示例:
# Generate sample arrays
a = np.random.randint(0, 5, size = (4, 3))
b = np.random.randint(-1, 6, size = (5, 3))
# Compare
a == b[:, None]
最后一行为您进行比较。输出数组将具有形状
(行数、行数、列数)
:在这种情况下,(5、4、3)。行的元素比较?
# Generate sample arrays
a = np.random.randint(0, 5, size = (4, 3))
b = np.random.randint(-1, 6, size = (5, 3))
# Compare
a == b[:, None]