*jdZddlmZddlmZddlZddlmZm Z erddl m Z  d dZ d dZ d d Zd d Zy)z Ops for masked arrays. ) annotations) TYPE_CHECKINGN)libmissing)nptc|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. 2Either `left` or `right` need to be an np.ndarray.ormethodT) kleene_or isinstancenpndarray TypeError raise_for_nan libmissingNAcopy zeros_like)leftright left_mask right_maskresult left_false right_falsemasks :C:\FundamentalWorkbench\vendor\pandas\core\ops\mask_ops.pyr r s8j)<< dBJJ 'LMM%% i'(  *+ * $Y& (:% '  4< $}}Y' 4< *--  "i/ 4<~~ 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. r xorr ) kleene_xorrrrrrrrr ones_liker)rrrrrrs rr#r#Qs<%z9== dBJJ 'LMM%& t$ JMM !<< *D 4< >>#D 4<:% 4<r c|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. Nr andr F) kleene_andrrrrrrrrr)rrrrrrrrs rr'r's6%z9== dBJJ 'LMM%& t$ JMM !I:%2D 4<>>#D~Q 4< i'(  *+ [L(Z:+-EF 4<r cxtj|r%tj|rt d|dyy)NzCannot perform logical 'z' with floating NaN)ris_floatrisnan ValueError)valuer s rrrs6 ||Erxx3F8;NOPP /r ) r%bool | np.ndarray | libmissing.NATyperr-rnp.ndarray | Nonerr.return3tuple[npt.NDArray[np.bool_], npt.NDArray[np.bool_]]) r%bool | libmissing.NAType | np.ndarrayrr1rr.rr.r/r0)r,objectr strr/None)__doc__ __future__rtypingrnumpyr pandas._libsrrrpandas._typingrr r#r'rr rr<s#  ": /: 0:!:" : 9 :z2 /2 02!2" 2 9 2j7 /7 07!7" 7 9 7tQr