#Initialize dictionaries and DataFrames for Plotting
favorability = {}
unfavorability = {}
favorability_fan = {}
unfavorability_fan = {}
df_seen = df.loc[df.loc[:,'seen_any'],:] #restrict to only respondents who have seen a star wars film
#____________________________________________________
#Compute unfavorability scores for fans and nonfans for each character
for character in vibe_cols:
favorability[character] = round(
sum(df_seen.loc[ #add up very favorably and somewhat favorably proportions
~df_seen.loc[:,'fan'],character] #filter to only nonfans
.value_counts(normalize=True)[
['very favorably', 'somewhat favorably']] #compute each proportion
.to_list())
,2) #round to two decimal places
unfavorability[character] = round( #same as above, for unfavorability of non fans
-1*sum(df_seen.loc[
~df_seen.loc[:,'fan'],character]
.value_counts(normalize=True)[
['very unfavorably', 'somewhat unfavorably']]
.to_list())
,2)
#___________________________________________________________
#Same as above, only filtering for fans instead of nonfans
favorability_fan[character] = round(
sum(df_seen.loc[
df_seen.loc[:,'fan'],character]
.value_counts(normalize=True)[
['very favorably', 'somewhat favorably']]
.to_list())
,2)
unfavorability_fan[character] = round(
-1*sum(df_seen.loc[
df_seen.loc[:,'fan'],character]
.value_counts(normalize=True)[
['very unfavorably', 'somewhat unfavorably']]
.to_list())
,2)
#_____________________________________________________________
#Codes for plotting
fig, ax = plt.subplots(figsize=(9, 5))
x = list(favorability.keys())
x_axis = np.arange(len(x)) #make numeric axis to be shifted
#______________________________________________________________
#generate the 4 horizontal bar plots
ax.barh(x_axis,favorability.values(), 0.4, label = 'General', color = '#2E67F8')
all_unfav = ax.barh(x_axis,unfavorability.values(), 0.4, label = 'General', color = '#FF3D4A')
ax.barh(x_axis,favorability_fan.values(), 0.4, label = 'Fan',left = 2.3, color = '#2E67F8')
ax.barh(x_axis,unfavorability_fan.values(), 0.4, label = 'Fan',left = 2.3, color = '#FF3D4A')
#______________________________________________________________
#Remove grid and tick labels
ax.grid(visible = False)
ax.set_xticklabels([])
ax.set_yticklabels([])
#______________________________________________________________
#Coordinates for placing character names
x_coords = {'Yoda': 1.07+0.2, 'Padme': 1.02+0.2, 'Jar jar': 1.05+0.2,
'R2D2': 1.05+0.2, 'C3PO': 1.05+0.2,
'Boba': 1.05+0.2, 'Lando': 1.02+0.2,
'Vader': 1.04+0.2, 'Palpatine': 0.96+0.2,
'Obi-Wan': 0.96+0.2, 'Anakin': 1+0.2,
'Leia': 1.05+0.2, 'Luke': 1.05+0.2,
'Han': 1.07+0.2 }
y_coord = 12.8
for y_label, x_coord in x_coords.items():
ax.text(x_coord, y_coord, y_label) #for each (name, coordinate) place name at coordinate
y_coord -= 1
#place vertical lines on either side of centered character names
ax.axvline(x= 0.2+0.9,ymin = .1, ymax = .9, alpha = .1, c = 'grey', lw = 1)
ax.axvline(x= 0.2+1.45,ymin = .1, ymax = .9, alpha = .1, c = 'grey', lw = 1)
#________________________________________________________________
#Horizontal line and labels on bottom of each plot
ax.axhline(-1, color='grey', linewidth=1, alpha=0.5,
xmin=-0.03, xmax=0.38)
ax.axhline(-1, color='grey', linewidth=1, alpha=0.5,
xmin=0.57, xmax=0.98)
ax.text(-0.5, -2, 'unfavorable'+ ' '*16 + 'favorable',
color='grey', alpha=0.5)
ax.text(1.7, -2, 'unfavorable'+ ' '*16 + 'favorable',
color='grey', alpha=0.5)
#__________________________________________________________________
#Horizontal line and labels on top of each plot
ax.axhline(15, color='grey', linewidth=1, alpha=0.9,
xmin=0.57, xmax=0.98)
ax.axhline(15, color='grey', linewidth=1, alpha=0.9,
xmin=-.03, xmax=0.4)
ax.text(0,15.5, 'NON FANS',weight = 'bold', alpha = .75)
ax.text(2.4,15.5,'FANS', weight = 'bold', alpha = .75)
#__________________________________________________________________
#Title and subtext explaining plot
ax.text(-0.3,19, 'Jar Jar: Disliked by General Viewers, Hated by Fans', weight = 'bold', size = 17, alpha = .75)
ax.text(-0.33,18, 'Percentage of group (non-fans or fans) who considered each character', alpha = .85)
ax.text(.5,17, ' unfavorable (red), and favorable (blue)', alpha = .85)
#___________________________________________________________________
#Place quantity near the largest quantities on each plot
#labeling unfavorability non fan values
for i, v in enumerate(unfavorability.values()):
if abs(v) >= .14:
ax.text(v-.2, i, str(v)[3:].lstrip('0') + '%', color='black', fontweight='bold', fontsize=10, ha='left', va='center')
else:
pass
#labeling unfavorability for fans
for i, v in enumerate(unfavorability_fan.values()):
if abs(v) >= .20:
ax.text(v + 2.1, i, str(v)[3:].lstrip('0') + '%', color='black', fontweight='bold', fontsize=10, ha='left', va='center')
else:
pass
#labeling favorability for non fans
for i, v in enumerate(favorability.values()):
if abs(v) >= .80:
ax.text(v+.05, i, str(v)[2:]+ '%', color='black', fontweight='bold', fontsize=10, ha='left', va='center')
else:
pass
#labeling favorability for fans
for i, v in enumerate(favorability_fan.values()):
if abs(v) >= .96:
ax.text(v + 2.35, i, str(v)[2:] + '%', color='black', fontweight='bold', fontsize=10, ha='left', va='center')
else:
pass
#______________________________________________________
plt.show()
