-
Notifications
You must be signed in to change notification settings - Fork 0
/
density.py
78 lines (68 loc) · 2.4 KB
/
density.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author: Ming
# @Date: 2019-07-11 23:57:09
# @Last Modified by: Ming
# @Last Modified time: 2019-07-12 10:53:46
import matplotlib as mpl
mpl.use('Agg')
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import click
CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'])
@click.command(context_settings=CONTEXT_SETTINGS)
@click.version_option(version='1.0')
@click.option('-i', '--input',
required=True,
type=click.Path(),
help="The table input to plot.")
@click.option('-n', '--names',
required=True,
help="The names group you want to plot, sep by ,")
@click.option('-p', '--prefix',
default='result',
help="The out prefix.")
@click.option('-x', '--xlab',
required=False,
help="The xlab for the plot")
@click.option('-y', '--ylab',
required=False,
help="The ylab for the plot.")
@click.option('-t', '--title',
required=False,
help="The title for the plot.")
@click.option('-c', '--colors',
required=False,
help="The colors used for plot, html code split by ,")
def cli(input, names, prefix, xlab, ylab, title, colors):
"""
Density plot with python.
"""
df = pd.read_csv(input, header=0, names=None, sep='\t')
names = names.strip().split(',')
colors = colors.strip().split(',') if colors else None
if colors and (len(names) != len(colors)):
raise ValueError('Diff name and color number!')
# draw
figure, ax = plt.subplots(figsize=(16, 8), dpi=300)
for i in range(len(names)):
if colors:
color = colors[i]
ax = sns.kdeplot(df[df.iloc[:, 0] == names[i]].iloc[:, 1],
shade=False,
color=color, label=names[i], alpha=.7)
else:
ax = sns.kdeplot(df[df.iloc[:, 0] == names[i]].iloc[:, 1],
shade=False,
label=names[i], alpha=.7)
if xlab:
ax.set_xlabel(xlab, fontsize=15)
if ylab:
ax.set_ylabel(ylab, fontsize=15)
if title:
ax.set_title(title, fontsize=22)
plt.savefig(prefix + '.svg', bbox_inches='tight')
plt.savefig(prefix + '.png', dpi=300, bbox_inches='tight')
if __name__ == "__main__":
cli()