Home Explore Help
Sign In
LiuFan
/
PrivacyScanData
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Branch: dev-lf
Branches Tags
PrivacyScanD...
/
target
/
visualize
/
visualize_5_26.py

visualize_5_26.py 2.7 KB
Permalink History Raw

123456789101112131415161718192021222324252627282930313233343536373839 
  1. def geneplot_mhat(df, markeridcol, chr, pv, gwasp, markernames, gfont, gstyle, ax):
    2. 

  2.     if markeridcol is not None:
    3. 

  3.         if markernames is not None and markernames is True:
    4. 

  4.             for i in df[markeridcol].unique():
    5. 

  5.                 if df.loc[df[markeridcol] == i, pv].iloc[0] <= gwasp:
    6. 

  6.                     if gstyle == 1:
    7. 

  7.                         plt.text(df.loc[df[markeridcol] == i, 'ind'].iloc[0], df.loc[df[markeridcol] == i, 'tpval'].iloc[0],
    8. 

  8.                                 str(i), fontsize=gfont)
    9. 

  9.                     elif gstyle == 2:
    10. 

  10.                         plt.annotate(i, xy=(df.loc[df[markeridcol] == i, 'ind'].iloc[0], df.loc[df[markeridcol] == i, 'tpval'].iloc[0]),
    11. 

  11.                                      xycoords='data', xytext=(5, -15), textcoords='offset points', size=6,
    12. 

  12.                                      bbox=dict(boxstyle="round", alpha=0.2),
    13. 

  13.                                      arrowprops=dict(arrowstyle="wedge,tail_width=0.5", alpha=0.2, relpos=(0, 0)))
    14. 

  14.         elif markernames is not None and isinstance(markernames, (tuple, list)):
    15. 

  15.             for i in df[markeridcol].unique():
    16. 

  16.                 if i in markernames:
    17. 

  17.                     if gstyle == 1:
    18. 

  18.                         plt.text(df.loc[df[markeridcol] == i, 'ind'].iloc[0], df.loc[df[markeridcol] == i, 'tpval'].iloc[0],
    19. 

  19.                             str(i), fontsize=gfont)
    20. 

  20.                     elif gstyle == 2:
    21. 

  21.                         plt.annotate(i, xy=(df.loc[df[markeridcol] == i, 'ind'].iloc[0], df.loc[df[markeridcol] == i, 'tpval'].iloc[0]),
    22. 

  22.                                      xycoords='data', xytext=(5, -15), textcoords='offset points', size=6,
    23. 

  23.                                      bbox=dict(boxstyle="round", alpha=0.2),
    24. 

  24.                                      arrowprops=dict(arrowstyle="wedge,tail_width=0.5", alpha=0.2, relpos=(0, 0)))
    25. 

  25.         elif markernames is not None and isinstance(markernames, dict):
    26. 

  26.             for i in df[markeridcol].unique():
    27. 

  27.                 if i in markernames:
    28. 

  28.                     if gstyle == 1:
    29. 

  29.                         plt.text(df.loc[df[markeridcol] == i, 'ind'].iloc[0], df.loc[df[markeridcol] == i, 'tpval'].iloc[0],
    30. 

  30.                              markernames[i], fontsize=gfont)
    31. 

  31.                     elif gstyle == 2:
    32. 

  32.                         plt.annotate(markernames[i], xy=(
    33. 

  33.                         df.loc[df[markeridcol] == i, 'ind'].iloc[0], df.loc[df[markeridcol] == i, 'tpval'].iloc[0]),
    34. 

  34.                                      xycoords='data', xytext=(5, -15), textcoords='offset points', size=6,
    35. 

  35.                                      bbox=dict(boxstyle="round", alpha=0.2),
    36. 

  36.                                      arrowprops=dict(arrowstyle="wedge,tail_width=0.5", alpha=0.2, relpos=(0, 0)))
    37. 

  37.     else:
    38. 

  38.         raise Exception("provide 'markeridcol' parameter")
    39. 

  39.