visuals_plotly.py 文件源码

def asset_bubble(output_by_assets):
    """Creates a crossfilter bokeh plot of results by asset

        :output_by_assets: Contains output by asset
        :type output_by_assets: dataframe
        :returns:
        :rtype:
    """
    import numpy as np
    import pandas as pd
    import plotly.plotly as py
    import plotly.graph_objs as go

    df_all = output_by_assets.copy()

    df = df_all[df_all['asset_category']!='Intellectual Property'].copy()

    # sort categories
    df['sort_order'] = df['asset_category']
    df['sort_order'].replace(asset_category_order,inplace=True)
    df.sort_values(by="sort_order",axis=0,ascending=True,inplace=True)
    df.reset_index(inplace=True)


    # update asset_category names for better printing
    df['asset_category'].replace(asset_categories_for_print,inplace=True)

    df.iplot(kind='bubble', x='metr_c', y='asset_category', size='assets', text='Asset',
             xTitle='Marginal Effective Tax Rate', yTitle='Asset Category',
             filename='BubbleChart.png')