首页 > 解决方案 > np.logical_or 与 reduce 返回不同的结果

问题描述

np.logical_orwithfunctools.reduce返回不同的结果。

kdf = pd.DataFrame(data={'col1' : [' False', 1, np.nan], 'dt': [datetime.now(), ' 2018-12-12', '2019-12-12'], 'bool': 
                         [False, True, True], 'i': [1,2,'3'], 'bnan': [False, True, np.nan], 'col2': [' True ', False, 'False']})

print([kdf[i].str.contains('^\s*F') for i in ['col1', 'col2']])

# [0    True
# 1     NaN
# 2     NaN
# Name: col1, dtype: object, 0    False
# 1      NaN
# 2     True
# Name: col2, dtype: object]

你可以看到这返回了预期的输出,但是当我们用 重用它时np.logical_or,它返回Nan第三行而不是True

from functools import reduce
reduce(np.logical_or, [kdf[i].str.contains('^\s*F') for i in ['col1', 'col2']])

# 0    True
# 1     NaN
# 2     NaN
# dtype: object

np.logical_or(np.nan, True)回报True。我希望reduce会将该函数应用于所有列表项,即

kdf['col1'].str.contains('^\s*F') | kdf['col2'].str.contains('^\s*F')

我错过了什么吗?

标签: pythonpandasnumpy

解决方案


我认为这是错误,对于使用 NaN 的正确处理应该将它们替换为一些 boolean ,例如False通过na=False参数:

from functools import reduce
a = reduce(np.logical_or, [kdf[i].str.contains('^\s*F') for i in ['col1', 'col2']])
print (a)
0    True
1     NaN
2     NaN
dtype: object

b = np.logical_or.reduce([kdf[i].str.contains('^\s*F') for i in ['col1', 'col2']])
print (b)
[True nan nan]

c = kdf['col1'].str.contains('^\s*F') | kdf['col2'].str.contains('^\s*F')
print (c)
0     True
1    False
2    False
dtype: bool

from functools import reduce
a = reduce(np.logical_or, [kdf[i].str.contains('^\s*F', na=False) for i in ['col1', 'col2']])
print (a)
0     True
1    False
2     True
dtype: bool

b = np.logical_or.reduce([kdf[i].str.contains('^\s*F', na=False) for i in ['col1', 'col2']])
print (b)
[ True False  True]

c = kdf['col1'].str.contains('^\s*F', na=False) | kdf['col2'].str.contains('^\s*F', na=False)
print (c)
0     True
1    False
2     True
dtype: bool

推荐阅读