a dY @sddlZddlmZddlZddZdgdefddZddZd d Z d d Z e d kre gdgdgdgdgdgZ gdZee e eddS)NcCs2|dd|f}t|}||dkr*|SdSdS)Nr)npargmax)matrixnodeparentsZ max_parentr5/mnt/chromeos/MyFiles/work/nlp_tsa/tsa/TSA/outline.py find_parents   r c Cstg}t||}|durp||vrp|||g|D]}|||gq0||krp|||t|||d|d|S)Nr )visited)r appendaddextend find_tree) rrdepthchildren max_depthr resultparentchildrrrrs  rcCs&d}|D]\}}||||9}q|SNr r)treerprobrrrrr find_probs rcCsi}tt|D]}||}g}tt|D]}||dkr,||q,t|||d}tdd|D}|rt||}t||vr|t||7<q||t|<qt|dddd} | d\} }i} | D]*\} } | | vr| | | q| g| | <q| |fS) Nr)rcSsg|] }t|qSr)tuple).0xrrr +z find_forests..cSs|dSrr)rrrr2rzfind_forests..T)keyreverse)rangelenr rrrsorteditems)rkZforestsirZ child_listjrrZsorted_forestsZforestrrrrrr find_forests"s,       r)c st|d\}}t||i}|D]}fdd||D||<q$d}g}t|D]>}|d|d}||D]} || |d| d}qvqX||fS) Nr csg|] }|qSrr)rr' sentencesrrrCrz#passage_outline..u 主题:u 主题: z-  )r)printkeysr$r ) rr+rr structureeachoutline outline_listr sentencerr*rpassage_outline=s     r5__main__)g޴2D•?gOfA?gQoN?gsɂ̤?)gqd?r7gCaO?g7?gTEWÏ?)gH 鍎?g ?r7gix#?g&o?)g $?f?g6:8?gt}?r7gϡ U1?)gM⤗?g/ɖ?g °?gz9@ε?r7)u 主题句子1u 主题句子2u 主题句子3u 主题句子4u 主题句子5)numpyrmatplotlib.pyplotpyplotpltr setrrr)r5__name__arrayrr+r.rrrrs