Unable to plot distribution of a column containing binary values using Python

Question

I'm trying to plot the original data before handling the imbalance in a way to show the class distribution and class imbalance (class is Failure =0/1) 2. I might need to do some transformation on the data in both cases to be able to visualize it.

Here's what the column looks like:

| failure |
|---------|
| 1       |
| 0       |
| 0       |
| 1       |
| 0       |

Here's what I have tried so far:

import numpy as np
from scipy.stats.kde import gaussian_kde

def distribution_scatter(x, symmetric=True, cmap=None, size=None):
    pdf = gaussian_kde(x)    
    w = np.random.rand(len(x))    

    if symmetric:        
        w = w*2-1    
        pseudo_y = pdf(x) * w    

    if cmap:        
        plt.scatter(x, pseudo_y, c=x, cmap=cmap, s=size)    

    else:        
        plt.scatter(x, pseudo_y, s=size)    

    return pseudo_y

Results:

The problem with the results:

I want the plot the distribution of 0's and 1's. For which I believe I need to transform it in someway.

Desired output:

perl · Accepted Answer · 2021-06-25 20:28:25Z

1

If you want a KDE plot, you can check kdeplot from seaborn:

x = np.random.binomial(1, 0.2, 100)
sns.kdeplot(x)

Output:

Update: Or a swarmplot if you want a scatter:

x = np.random.binomial(1, 0.2, 25)
sns.swarmplot(x=x)

Output:

Update 2: In fact, your function seems to also produce a reasonable visualization:

distribution_scatter(np.random.binomial(1, 0.2, 100))

Output:

edited Jun 25, 2021 at 20:28

answered Jun 25, 2021 at 20:07

perl

9,9811 gold badge14 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

7 Comments

Qasim Khan Over a year ago

Thanks for mentioning! But I want a scatter plot which will show the distribution of 0's and 1's.

perl Over a year ago

@QasimKhan Maybe swarmplot (see Update)?

perl Over a year ago

@QasimKhan And actually the function that you currently have seems to be OK as well (see Update 2). Can you explain in a bit more detail what specifically you're trying to achieve?

Qasim Khan Over a year ago

Hey @perl, please check the question again, I added a desired output section.

perl Over a year ago

@QasimKhan What are the X and Y values in your chart? Do you want them generated somehow?

|

Collectives™ on Stack Overflow

Unable to plot distribution of a column containing binary values using Python

Here's what the column looks like:

Here's what I have tried so far:

Results:

The problem with the results:

Desired output:

1 Answer 1

7 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

Here's what the column looks like:

Here's what I have tried so far:

Results:

The problem with the results:

Desired output:

1 Answer 1

7 Comments

Your Answer

Sign up or log in

Post as a guest

Related