U md~@sddlmZmZddlZddlZddlmZ ddlm Z ddl m Z ddl mZdee ejejfeeeeeedfed d d Zdejeeeeeedfd ddZdS))OptionalUnionN)pyplot)rcParams)AnnData)_utilsFT)adata_or_resultlogshowsavehighly_variable_genesc Cst|tr$|j}|jdddk}n |}t|tjr@d|jk}nd}|rP|j}n|j}|j }|rn|j } |j } n |j } |j } td} tjd| d| d fd tjd d t| | gD]>\} } td d| d td dgddg||gD]0\}}}||| |}}tj||||d dq|rtdtdt| }|dkrTd|nd}tdt|dt|t|dt| | dkrttdrdndd|sdnd}tdrdndd || dkrdndqt!j"d||d|dkrt#Sd S)!a.Plot dispersions or normalized variance versus means for genes. Produces Supp. Fig. 5c of Zheng et al. (2017) and MeanVarPlot() and VariableFeaturePlot() of Seurat. Parameters ---------- adata Result of :func:`~scanpy.pp.highly_variable_genes`. log Plot on logarithmic axes. show Show the plot, do not return axis. save If `True` or a `str`, save the figure. A string is appended to the default filename. Infer the filetype if ending on {{`'.pdf'`, `'.png'`, `'.svg'`}}. ZhvgZflavorZ seurat_v3variances_normFzfigure.figsizerr)Zfigsizeg333333?)Zwspacezhighly variable genesz other genesblackZgrey)labelcsr gffffff?g?g?z $log_{10}$ zmean expressions of genes dispersions variancesz {} of genesz (normalized)z (not normalized)filter_genes_dispersion)r r N)$ isinstancervarZunspd DataFramecolumnsZhighly_variable gene_subsetmeansrrrZdispersions_normrplZfigureZsubplots_adjust enumerateZsubplotzipnpZscatterZxscaleZyscaleminZxlimmaxZylimZlegendZxlabelZylabelformatrZsavefig_or_showZgca)r r r r r resultZseurat_v3_flavorrrZ var_or_dispZvar_or_disp_normsizeidxdrcolormaskZmeans_Z var_or_disps_Zy_minZ data_typer,W/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scanpy/plotting/_preprocessing.pyr sb           r r&r r r cCst||||dddS)a Plot dispersions versus means for genes. Produces Supp. Fig. 5c of Zheng et al. (2017) and MeanVarPlot() of Seurat. Parameters ---------- result Result of :func:`~scanpy.pp.filter_genes_dispersion`. log Plot on logarithmic axes. show Show the plot, do not return axis. save If `True` or a `str`, save the figure. A string is appended to the default filename. Infer the filetype if ending on {{`'.pdf'`, `'.png'`, `'.svg'`}}. F)r r r r N)r r.r,r,r-rcsr)FNNT)FNN)typingrrnumpyr"ZpandasrZ matplotlibrrrZanndatarrrrZrecarrayboolstrr rr,r,r,r-s4      V