Spaces:
Sleeping
Sleeping
File size: 1,569 Bytes
b07caec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
# ===================
# Part 1: Importing Libraries
# ===================
import matplotlib.pyplot as plt
import numpy as np
np.random.seed(0)
# ===================
# Part 2: Data Preparation
# ===================
# Data
emotions = ["Ang", "Cnt", "Dis", "Fea", "Joy", "Ntr", "Sad", "Sur"]
negative = [500, 300, 200, 1000, 0, 0, 500, 0]
positive = [0, 0, 0, 0, 2000, 0, 0, 0]
none = [0, 0, 0, 0, 0, 3500, 0, 0]
mixed = [0, 0, 0, 0, 0, 0, 0, 300]
labels = ["negative", "positive", "none", "mixed"]
xlabel = "Emotions"
ylabel = "Frequency"
ylim = [0, 4000]
# ===================
# Part 3: Plot Configuration and Rendering
# ===================
# Set figure size to match the original image's dimensions
plt.figure(figsize=(6, 4))
# Plotting
bar_width = 0.8
index = np.arange(len(emotions))
plt.bar(index, negative, bar_width, color="red", label=labels[0])
plt.bar(index, positive, bar_width, color="green", label=labels[1], bottom=negative)
plt.bar(
index,
none,
bar_width,
color="grey",
label=labels[2],
bottom=[i + j for i, j in zip(negative, positive)],
)
plt.bar(
index,
mixed,
bar_width,
color="orange",
label=labels[3],
bottom=[i + j + k for i, j, k in zip(negative, positive, none)],
)
# Labels and Title
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.ylim(ylim)
# plt.title('Emotion Frequencies by Sentiment')
plt.xticks(index, emotions)
plt.legend(loc="upper left")
# ===================
# Part 4: Saving Output
# ===================
# Show plot
plt.tight_layout()
plt.savefig("bar_26.pdf", bbox_inches="tight")
|