File tree 1 file changed +12
-2
lines changed
1 file changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -54,7 +54,12 @@ def mask_missing(arr, values_to_mask):
54
54
if is_numeric_v_string_like (arr , x ):
55
55
mask = False
56
56
elif arr .dtype .kind == 'O' :
57
- mask = np .array ([i == x .tolist () for i in arr .ravel ()])
57
+ try :
58
+ m = [i == x .tolist () for i in arr .ravel ()]
59
+ except AttributeError :
60
+ m = [i == x for i in arr .ravel ()]
61
+ finally :
62
+ mask = np .array (m , dtype = bool )
58
63
else :
59
64
mask = arr == x
60
65
@@ -68,7 +73,12 @@ def mask_missing(arr, values_to_mask):
68
73
if is_numeric_v_string_like (arr , x ):
69
74
mask |= False
70
75
elif arr .dtype .kind == 'O' :
71
- mask |= np .array ([i == x .tolist () for i in arr .ravel ()])
76
+ try :
77
+ m = [i == x .tolist () for i in arr .ravel ()]
78
+ except AttributeError :
79
+ m = [i == x for i in arr .ravel ()]
80
+ finally :
81
+ mask |= np .array (m , dtype = bool )
72
82
else :
73
83
mask |= arr == x
74
84
You can’t perform that action at this time.
0 commit comments