h!|dZddlmZddlZddlmZmZ d dZ d dZ d dZ d dZ y) z Ops for masked arrays. ) annotationsN)libmissingc|t||||St|tjs t dt |d|t jur|j}n||z}|!||z}||z}||z||zz||zz}||fS|durtj|}||fS|t jur||z|z}||fS|j}||fS)a Boolean ``or`` using Kleene logic. Values are NA where we have ``NA | NA`` or ``NA | False``. ``NA | True`` is considered True. Parameters ---------- left, right : ndarray, NA, or bool The values of the array. left_mask, right_mask : ndarray, optional The masks. Only one of these may be None, which implies that the associated `left` or `right` value is a scalar. Returns ------- result, mask: ndarray[bool] The result of the logical or, and the new mask. 1Either `left` or `right` need to be a np.ndarray.ormethodT) kleene_or isinstancenpndarray TypeError raise_for_nan libmissingNAcopy zeros_like)leftright left_mask right_maskresult left_false right_falsemasks =D:\jyotish\venv\Lib\site-packages\pandas/core/ops/mask_ops.pyr r s8j)<< dBJJ 'KLL%% i'(  *+ * $Y& (:% '  4< D===+D 4< jmm #EYJ&)3D 4<>>#D 4<ct|t||||St|tjs t dt |d|t jurtj|}n||z }|?|t jurtj|}||fS|j}||fS||z}||fS)a' Boolean ``xor`` using Kleene logic. This is the same as ``or``, with the following adjustments * True, True -> False * True, NA -> NA Parameters ---------- left, right : ndarray, NA, or bool The values of the array. left_mask, right_mask : ndarray, optional The masks. Only one of these may be None, which implies that the associated `left` or `right` value is a scalar. Returns ------- result, mask: ndarray[bool] The result of the logical xor, and the new mask. rxorr ) kleene_xorr r rrrrrr ones_liker)rrrrrrs rr!r!Ls<%z9== dBJJ 'KLL%& t$ JMM !<< *D 4< >>#D 4<:% 4<rc|t||||St|tjs t dt |d|t jurtj|}n||z}|<|t jur ||z|z}||fS|j}|durd|dd||fS||z}||z}||z||zz}||fS)a Boolean ``and`` using Kleene logic. Values are ``NA`` for ``NA & NA`` or ``True & NA``. Parameters ---------- left, right : ndarray, NA, or bool The values of the array. left_mask, right_mask : ndarray, optional The masks. Only one of these may be None, which implies that the associated `left` or `right` value is a scalar. Returns ------- result, mask: ndarray[bool] The result of the logical xor, and the new mask. Nrandr F) kleene_andr r rrrrrrr)rrrrrrrrs rr%r%s6%z9== dBJJ 'KLL%& t$ JMM !I:%2D 4<>>#D~Q 4< i'(  *+ [L(Z:+-EF 4<rcxtj|r%tj|rt d|dyy)NzCannot perform logical 'z' with floating NaN)ris_floatr isnan ValueError)valuer s rrrs6 ||Erxx3F8;NOPP /r)r%bool | np.ndarray | libmissing.NATyperr+rnp.ndarray | Nonerr,)r%bool | libmissing.NAType | np.ndarrayrr-rr,rr,)r strreturnNone) __doc__ __future__rnumpyr pandas._libsrrrr r!r%rrrr6s# ; /; 0;!;" ;|2 /2 02!2" 2j7 /7 07!7" 7tQr