HR-eP@dZddlZddlZddlZddZddZdZddZdS) zA Collection of useful miscellaneous functions. misc.py: Collection of useful miscellaneous functions. :Author: Hannes Breytenbach (hannes@saao.ac.za) Nctjd}nfd}tt|t ||}|r|dn|S)zKfind the first index position for the which the callable pred returns True.Nc&|dS)Nrxpreds 5lib/python3.11/site-packages/astropy/io/ascii/misc.pyz"first_true_index..sadr)operator itemgetternextfilter enumerate)iterabler defaultfunciis ` r first_true_indexrs\ |"1%%#### fT9X..// 9 9B #2a55G#r cN tj}nfd}t|||S)zLfind the first index position for the which the callable pred returns False.Nc| SNrrs r r z#first_false_index..sTT!WWr )r not_r)rr rrs ` r first_false_indexrs1 |}$$$$ HdG 4 44r ct|d}t|s|S|d|d rdnddttjjstdttjjrfd }nWttrd Dfd }n/td tttt|| }d|vr*|ddrt|}tt!tt|S)aV Sorts any number of lists according to: optionally given item sorting key function(s) and/or a global sorting key function. Parameters ---------- One or more lists Keywords -------- globalkey : None revert to sorting by key function globalkey : callable Sort by evaluated value for all items in the lists (call signature of this function needs to be such that it accepts an argument tuple of items from each list. eg.: ``globalkey = lambda *l: sum(l)`` will order all the lists by the sum of the items from each list if key: None sorting done by value of first input list (in this case the objects in the first iterable need the comparison methods __lt__ etc...) if key: callable sorting done by value of key(item) for items in first iterable if key: tuple sorting done by value of (key(item_0), ..., key(item_n)) for items in the first n iterables (where n is the length of the key tuple) i.e. the first callable is the primary sorting criterion, and the rest act as tie-breakers. Returns ------- Sorted lists Examples -------- Capture sorting indices:: l = list('CharacterS') In [1]: sortmore( l, range(len(l)) ) Out[1]: (['C', 'S', 'a', 'a', 'c', 'e', 'h', 'r', 'r', 't'], [0, 9, 2, 4, 5, 7, 1, 3, 8, 6]) In [2]: sortmore( l, range(len(l)), key=str.lower ) Out[2]: (['a', 'a', 'C', 'c', 'e', 'h', 'r', 'r', 'S', 't'], [2, 4, 0, 5, 7, 1, 3, 8, 9, 6]) r globalkeykeyNcdSNrrrs r r zsortmore..\Ar c|Srrr!s r r zsortmore.._r"r cdSr rr!s r r zsortmore..asqr zglobalkey needs to be callablec0||dfSr rrrrs r r zsortmore..gsyy!}cc!A$ii0r c3$K|] }|r|ndV dS)cdSr rr!s r r z$sortmore...isQr Nr).0ks r zsortmore..is-441A&qq;;444444r c`|ftdt|DzS)Nc32K|]\}}||VdSrr)r)fzs r r+z-sortmore....js..N.NAqqtt.N.N.N.N.N.Nr )tuplezipr&s r r zsortmore..js4yy!}&.N.N#c1++.N.N.N)N)NNr zIkw arg 'key' should be None, callable, or a sequence of callables, not {})rorder)descendreverse)listlenget isinstance collectionsabcCallable ValueErrorr0KeyErrorformattypesortedr1 startswithreversedmap)argskwfirstr*resrrs @@r sortmorerH#s` aMME u:: {##I &&--C {  +CC+C L i!9 : :;9:::#{/00  0 0 0 0 0 C   44444 N N N N N W ^ ^S      c4j!!q ) ) )C"}} g; ! !"8 9 9 3--C T39%% & &&r csdt|di}fd}tjt||}d|D}|S)zMExtends the itertools.groupby functionality to arbitrary number of iterators.c|Srrr!s r r zgroupmore..}sr rc&|dSr r)rrs r r zgroupmore..sdd1Q4jjr c32K|]\}}|t|fVdSr)r1)r)rgroupss r r+zgroupmore..s/>> Vc6l#>>>>>>r )rH itertoolsgroupbyr1)ritsnfunczipperunzippers` r groupmorerTzs` { C "T " "C E  sCy% 0 0F>>v>>>H Or )NNr) __doc__collections.abcr9rNr rrrHrTrr r rWs$$$$5555T'T'T'nr