how to plot class labels using a distribution list

Question

I have a dataset with train and test sets and three classes A,B,and C. I want to create a plot in which I show the distribution of data labels in each class for TRAIN and TEST sets separately (these are binary class labels 0 and 1). Ideally, I would like to show TRAIN and TEST stats in different colours, maybe in a bar chart. These are the values:

a_train = [40,75]
a_test = [10,19]

b_train=[41,75]
b_test=[10,19]

c_train=[51,75]
c_test=[12,19]

I have tried to use pyplot but was confused how to create the plot:

import numpy as np                                                               
import matplotlib.pyplot as plt

top=[(['A',[[40,75],[10,19]]]),('B',[[41,75],[10,19]]),('C',[[51,75],[12,19]])]

labels, ys = zip(*top)
xs = np.arange(len(labels)) 
width = 1

plt.bar(xs, ys, width, align='center')

plt.xticks(xs, labels) 
plt.yticks(ys)

which gives this error:

ValueError: shape mismatch: objects cannot be broadcast to a single shape

Take a look on this thread :- stackoverflow.com/questions/33203645/… — Syed Ahmad
– Syed Ahmad, Commented Jun 22, 2021 at 11:29

MD Mushfirat Mohaimin · Accepted Answer · 2021-06-22 15:15:02Z

1

labels = ['a_train', 'a_test', 'b_train', 'b_test','c_train','c_test']
Positive = [40,         10,        41,      10,       51,       12]
Negative = [75,         19,        75,      19,       75,       19]

x = np.arange(len(labels))
width = 0.30  # the width of the bars

fig, ax = plt.subplots()
rects1 = ax.bar(x - width/2, Positive, width, label='Positive')
rects2 = ax.bar(x + width/2, Negative, width, label='Negative')

ax.set_ylabel('Values')
ax.set_xticks(x)
ax.set_xticklabels(labels)
ax.legend()

plt.show()

Result

answered Jun 22, 2021 at 15:15

MD Mushfirat Mohaimin

2,0724 gold badges14 silver badges24 bronze badges

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

Collectives™ on Stack Overflow

how to plot class labels using a distribution list

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related