Python: 根据两个数组中的值创建类别

Question

假设我有两个列表：

arrayA = np.array([3,4,1,2,5,6,8,6,3])
arrayB = np.array([4,2,5,6,1,3,6,5,3])

基本上表示二维中的一个点。

我想要一个如下所示的标签列表：

listLael = [type1,type2,type1,type2,...]

与 arrayA 和 arrayB 具有相同长度并且

type1 if arrayA value >= 5 and arrayB value >= 5
type2 if eith arrayA or arrayB value < 5

我知道我可以遍历两个数组并获取它，但是有没有快速方便的方法来使用 numpy 数组？

Answer 1

使用numpy.where:

>>> np.where((arrayA >= 5) & (arrayB >= 5), 'type1', 'type2')
array(['type2', 'type2', 'type2', 'type2', 'type2', 'type2', 'type1',
       'type1', 'type2'],
      dtype='|S5')

Python: 根据两个数组中的值创建类别

Python: create category based on values in two arrays

python

arrays

label

numpy