hMhp ddlmZddlZddlZdeejeejfdeejeejfdej fdZ ddej d e defd Z d ej d ej dej fd Z dS))UnionNxyreturnct|tjr|jnd}t|tjr|jnd}tjt j||gj||gS)zConcat two array-like data structures into a DataFrame :param x: pd.Series, list or np.ndarray :param y: pd.Series, list or np.ndarray :return: pd.DataFrame rr)columns) isinstancepdSeriesname DataFramenparrayT)rrx_namey_names J/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/phik/utils.pyarray_like_to_dataframersg"!RY// 8QVVSF!!RY// 8QVVSF <!Q((*VV4D E E EEFdfverbosec$|tjgj}|r1|r/t dt|t|tstd|S)zSelect columns that have a dtype part of np.number :param df: DataFrame :param bool verbose: print all interval columns that are guessed :return: list of interval columns )includez&interval columns not set, guessing: {}z Could not guess interval columns) select_dtypesrnumberrtolistprintformatstrr list ValueError)rr interval_colss rguess_interval_colsr#s$$bi[$99AHHJJMSS 6==c->P>PQQRRR mT * *=;<<< robservedexpectedc6|j}|j}tt|t|z }tt|t|z }|j}|j}tt|t|z }tt|t|z } |D]} d|| <|D]} d|| <t |j|_t |j|_t |jt |jksJ|D]0} tjt |j|j | <1| D]0} tjt |j|j | <1t |jt |jksJ|S)aWMake observed and expected (pandas) histograms equal in shape Sometimes expected histogram shape need filling / pruning to make its shape equal to observed. Give expected the same number of columns and rows. Needed for proper histogram comparison. :param pd.DataFrame observed: The observed contingency table. The table contains the observed number of occurrences in each cell. :param pd.DataFrame expected: The expected contingency table. The table contains the expected number of occurrences in each cell. :return: expected frequencies, having the same shape as observed g) rrr setindexsortedlenrzerosloc) r$r%o_colse_colso_cols_missinge_cols_missingo_idxe_idx o_idx_missing e_idx_missingcis rmake_shapes_equalr7&s  $ $ & &F   $ $ & &F#f++F 344N#f++F 344N N ! ! # #E N ! ! # #EUc%jj011MUc%jj011M  h.//Hh.//H x C(8$9$9 9 9 9 9::(3x'7#8#899 Q ::(3x'7#8#899 Q x~  #hn"5"5 5 5 5 5 Or)F)typingrpandasr numpyrr r ndarrayr rboolr#r7rrr>s F RYbj () F.3BItRZ4O.P F\ F F F F  BL 4 D     ' ' '''''''r