温馨提示×

numpy怎么比较两个数组大小

小亿
302
2023-09-28 20:56:31
栏目: 编程语言

在NumPy中,可以使用比较运算符(如<、>、<=、>=、==、!=)对两个数组进行逐元素比较,并生成一个布尔数组表示比较结果。例如,要比较两个数组a和b的大小,可以使用以下代码:

import numpy as np
a = np.array([1, 2, 3])
b = np.array([2, 3, 4])
# 比较两个数组的大小
result = a < b
print(result)  # 输出:[ True  True  True]

在这个例子中,数组a的每个元素都与数组b的对应元素进行比较。如果a中的元素小于b中的相应元素,则结果数组中的对应位置为True,否则为False。

你也可以使用其他比较运算符进行比较,比如a > b, a <= b, a >= b, a == b, a != b等。

如果要比较多维数组的大小,可以使用相同的方法。只需确保两个数组的形状相同,即对应维度的元素个数相等。

0