Question

我有2个numpy数组：

xarr = np.array([1.1, 1.2, 1.3, 1.4, 1.5])
y = np.array([1.1,1.2])

我想检查xarr的每个元素是属于y还是等于1.3。如果一个元素属于y，则返回＆＃34; y＆＃34;，如果一个元素等于1.3，则返回＆＃34; y1＆＃34;否则返回＆＃34; n＆＃34;

我试过了：

x = np.where(xarr in y,"y",np.where(xarr == 1.3,"y1","n"))

但是得到了错误的结果，前2个元素应该是＆＃34; y＆＃34;而不是＆＃34; n＆＃34;

['n' 'n' 'y1' 'n' 'n']

不知道我做错了什么。非常感谢任何帮助

Answer 1

你可以使用numpy.in1d，其余的很简单：

关键部分：

In [25]: np.in1d(xarr, y)
Out[25]: array([ True,  True, False, False, False], dtype=bool)

整个例子：

In [16]: result = np.empty(len(xarr), dtype=object)

In [17]: result
Out[17]: array([None, None, None, None, None], dtype=object)

In [18]: result.fill("n")

In [19]: result
Out[19]: array(['n', 'n', 'n', 'n', 'n'], dtype=object)

In [20]: result[np.in1d(xarr, y)] = 'y'

In [21]: result
Out[21]: array(['y', 'y', 'n', 'n', 'n'], dtype=object)

In [23]: result[xarr == 1.3] = 'y1'

In [24]: result
Out[24]: array(['y', 'y', 'y1', 'n', 'n'], dtype=object)

编辑：

对原始尝试的一个小修改：

In [16]: x = np.where(np.in1d(xarr, y),"y",np.where(xarr == 1.3,"y1","n"))

In [17]: x
Out[17]: 
array(['y', 'y', 'y1', 'n', 'n'], 
      dtype='|S2')

原始尝试中的问题是xarr in y仅提供False。

Numpy - 检查数组的元素是否属于另一个数组

1 个答案: