ó jÜxec@ sfddlmZddlZddlZddlmZmZede d„ƒZ ed„ƒZ dS(iÿÿÿÿ(tdivisionN(tcheckt handle_errorsc C s tt|tjƒdt|ƒƒ|jƒ}t|dddhkd|ƒtt|tƒdt|ƒƒ|s—ttj |j ƒj ƒdƒnt|j ddkd ƒt|j ddkd ƒxt |jƒD]n\}}t|ƒ}tt|tƒd ||fƒtt|ƒdkd |t|ƒt|ƒfd ƒqÛWt|jƒ}|jƒ||}d|j_y|jjtƒ}Wntk r²ttdƒnXtt ||jkƒdƒttt|jƒƒt|jƒkdƒ|dkr+tt |j jƒdkƒdƒnÚ|dkrött |j jƒdkƒdƒ|jddƒj }tttj|dƒƒ dƒt tj|dƒƒsdGH|j |dd…tjf|j dd…dd…f<|jƒ}qn|dkrn|S(sð Checks to make sure that the input dataframe, df, represents a valid matrix, i.e., an object that can be displayed as a logo. parameters ---------- df: (dataframe) A pandas dataframe where each row represents an (integer) position and each column represents to a (single) character. matrix_type: (None or str) If 'probability', validates df as a probability matrix, i.e., all elements are in [0,1] and rows are normalized). If 'information', validates df as an information matrix, i.e., all elements >= 0. allow_nan: (bool) Whether to allow NaN entries in the matrix. returns ------- out_df: (dataframe) A cleaned-up version of df (if possible). s<out_df needs to be a valid pandas out_df, out_df entered: %st probabilityt informations?matrix_type = %s; must be None, "probability", or "information"s+allow_nan must be of type bool; is type %s.sFsome matrix elements are not finite. Set allow_nan=True to allow this.iis&df has zero rows. Needs multiple rows.s,df has zero columns. Needs multiple columns.s-column number %d is of type %s; must be a strs#column %d is %s and has length %d; smust have length 1.tposs_could not convert df.index to type int. Check that all positions have integer numerical values.srcould not convert df.index values to int without changingsome values. Make sure that df.index values are integers.s@not all values of df.index are unique. Make sure all are unique.snot all values in df are >=0.taxisgs&some columns in df sum to nearly zero.gð?sMin validate_matrix(): Row sums in df are not close to 1. Reormalizing rows...N(!Rt isinstancetpdt DataFramettypetcopytNonetbooltnptisfinitetvaluestalltshapet enumeratetcolumnststrtlentreprtlisttsorttindextnametastypetintt TypeErrortFalsetsettraveltsumtanytisclosetnewaxistloc( tdft matrix_typet allow_nantout_dftitcolt char_colst int_indextsums((s;/tmp/pip-install-l3LICk/logomaker/logomaker/src/validate.pytvalidate_matrixsb         $ % 9 cC sÆt|dtƒ}tt|jjƒdkƒdƒ|jddƒj}tttj |dƒƒ dƒttj |dƒƒsÂd GH|j|d d …tj f|j d d …d d …f=0.Rigs+some columns in prob_df sum to nearly zero.gð?sVin validate_probability_mat(): Row sums in df are not close to 1. Reormalizing rows...N( R0RRRRR!R"R#RR$R%R&(R'tprob_dfR/((s;/tmp/pip-install-l3LICk/logomaker/logomaker/src/validate.pytvalidate_probability_mat„s<( t __future__RtnumpyRtpandasRtlogomaker.src.error_handlingRRR RR0R2(((s;/tmp/pip-install-l3LICk/logomaker/logomaker/src/validate.pyts   |