Package: bubble-plot
bubble_plotHi everyone!I love data visualizations! And if you love them too, I think you will find this bubble plot very nice and useful.How to installMotivation & UsageUsage ExampleUsage Example 2DependenciesContactHow to installVery simple - just write in your command line:pipinstallbubble_plotMotivation & UsageThe goal for the bubble plot is to help us visualize linear and non-linear connections between numerical/categorical features in our data in an easy and simple way. The bubble plot is a kind of a 2-dimensional histogram using bubbles. It suits every combination of categorical and numerical features.The bubble size is proportional to the frequency of the data points in this point.Function signature:bubble_plot(df,x,y,z_boolean=None,ordered_x_values=None,ordered_y_values=None,bins_x=10,bins_y=10,fontsize=16,figsize=(15,10),maximal_bubble_size=5000,normalization_by_all=False,log=False)For numerical features the values will be presented in buckets (ten equally spaced bins will be used as default, you can provide the specific bins / bins number through thebin_xandbins_yparameters).For categorical features the features will be presented according to their categories. If you would like a specific order for the categories presentation please supply a list of the values by order using theordered_x_values/ordered_y_valuesparameters.You can plot a numerical feature vs. another numerical feature or vs. a categorical feature or a categorical feature vs another categorical feature or numerical feature. All options are possible.Setting the parameter normalization_by_all to False defines that we would like to plot P(y/x), meaning, plot the distribution of y given x. Each column in this plot is an independent (1D) histogram of the values of the y given x. Setting the parameter normalization_by_all to True would plot the joint distribution of x and y, P(x,y), this is in fact a 2D histogram with bubbles.Setting thelogparameter toTruewould apply the natural log function - element wise - on the counts which will make the differences between the largest bubble to the smallest bubble much smaller, so if you have large differences between the frequencies of different values you might want to use that.Setting thez_booleanparameter to a name of categorical field with two categories / boolean field would make the color of the bucket  be proportional to the ratio ( (boolean_z==value_1).sum()/(boolean_z==value_1).sum() + (boolean_z==value_2).sum()) of the z values for this bucket.Usage Exampleimportpandasaspdfrombubble_plot.bubble_plotimportbubble_plotfromsklearn.datasetsimportload_bostonimportseabornassnssns.set_style("darkgrid")data=load_boston()df=pd.DataFrame(columns=data['feature_names'],data=data['data'])df['target']=data['target']bubble_plot(df,x='RM',y='target')The resulting bubble plot will look like this:Usage Example 2Census income dataset - plot the age vs. hours per week vs. the income level.
How is that even possible? Can we visualize three dimensions of information in a two dimensional plot?importpandasaspdimportseabornassnsfrombubble_plot.bubble_plotimportbubble_plotsns.set_style("darkgrid")df=pd.read_csv("adult.csv")bubble_plot(df,x='age',y='hours-per-week',z_boolean='target')The resulting bubble plot will look like this:P(x,y), x: age, y: working hours, color — proportional to the rate of high income people within each bucketIn this bubble plot, we see  the joint distribution of the hours-per-week vs. the age (p(x,y)), but here the color is proportional to the rate of high income people — (#>50K/((#>50K)+(#≤50K)) - within all the people in this bucket . By supplying the z_boolean variable, we added additional dimension to the plot using the color of the bubble.The pinker the color, the higher the ratio for the given boolean feature/target Z. See colormap in the image.Cool colormap — Pink would stand for the higher ratios in our case, cyan would stand for the lower ratiosThis plot shows us clearly that the higher income is much more common within people of age higher than 30 which work more than 40 hours a week.DependenciespandasnumpymatplotlibContactMore usage examples and explanations can be found at:https://medium.com/@DataLady/exploring-the-census-income-dataset-using-bubble-plot-cfa1b366313bPlease let me know if you have any questions. My email ismeir.shir86@gmail.com.Enjoy,
Shir