nan_policy ='omit' 在 scipy.stats.mode() 中不起作用

如何解决nan_policy ='omit' 在 scipy.stats.mode() 中不起作用

我正在尝试查找存储在 b 中的返回次数最多的标签，使用 a 作为布尔数组来确定要用作哪些标签：

a = np.array([[1.0,0.,0.],[1.0,1.]])
b = np.array(['a','b','a'])
x = np.where(a,b,np.nan)
x
# returns:
# array([['a','nan','nan'],#        ['a','a']],dtype='<U32')

为了找到模式，调用了 scipy.stats.mode()：

import scipy.stats
lb,_ = scipy.stats.mode(x,axis = 1,nan_policy ='omit')
lb

它回来了

array([['nan'],['a']],dtype='<U32')

但是，如果我正确理解“省略”，则不应返回“nan”的任何结果。我想忽略所有 nan。

我已检查 scipy 版本为 1.14.1。

** 更新： 可能是数组 b 由字符串而不是 int/float 组成的问题。 除了索引 b 并稍后按索引检索 b 中的标签之外，还有其他解决方案吗？