In [1]:

import sys
sys.path.insert(0, '../')

import dotplot
import dotplot.utils
import pandas as pd

%config InlineBackend.figure_format = 'retina'

In [2]:

columns, index = ['sample1', 'sample2', 'sample3'], ['A_1', 'B_1', 'A_2', 'B_2', 'A_3','B_3']
df_sizes = pd.DataFrame([[0.6, 1.5, 0.7], [1.3, .9, 1.6], [2.8, 1.0, .5], [2.1, 1.7, 1.85], [2.1, 1.7, 1.85], [1.2, .7, 1.9]], columns=columns, index=index)
df_colors = pd.DataFrame([[0.6, 4, 6], [3, .9, 1.6], [2.8, 5, .5], [2.1, .1, 1.85], [2.1, 1.7, 1.85], [1.2, .8, 1.9]], columns=columns, index=index)
df_circles = pd.DataFrame([[0.7, 1.4, 0.9], [1.3, .75, 1.66], [2.5, 1.3, .7], [2.2, 1.7, 1.85], [2.1, 1.7, 1.85], [1.45, .99, 1.9]], columns=columns, index=index)

df_row_colors = pd.DataFrame({'feature_category':['r', 'r', 'b', 'b', 'b', 'r']}, index=index)
df_col_colors = pd.DataFrame({'gender':['#FFB6C1', '#87CEFA', '#87CEFA']}, index=columns)


masks = pd.DataFrame([['group A']*3, ['group B']*3, ['group A']*3, ['group B']*3, ['group A']*3, ['group B']*3], columns=columns, index=index)

示例1¶

In [3]:

dp = dotplot.DotPlot(df_size=df_sizes, df_color=df_colors, df_circle=df_circles,
                     row_colors=df_row_colors, col_colors=df_col_colors)
_ = dp.plot(size_factor=50, cmap='Spectral_r')

In [4]:

_ = dp.plot(size_factor=50, cmap='Spectral_r', cluster_row=True, cluster_col=True)

示例2¶

In [5]:

dp = dotplot.DotPlot(df_size=df_sizes, df_circle=df_circles, row_colors=df_row_colors, col_colors=df_col_colors, mask_frames=masks)
_ = dp.plot(size_factor=50, cmap='Spectral_r')

In [6]:

dp = dotplot.DotPlot(df_size=df_sizes, df_circle=df_circles, df_color=df_colors, row_colors=df_row_colors, col_colors=df_col_colors, mask_frames=masks)
_ = dp.plot(size_factor=50, cmap='Spectral_r')