Thanks to visit codestin.com
Credit goes to github.com

Skip to content

DOC: Update multiple category bar chart examples #24498

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 16 commits into from
Dec 11, 2022
Merged

DOC: Update multiple category bar chart examples #24498

Show file tree
Hide file tree
Changes from 7 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
97a9664
change category bar chart examples to use penguin data
kostyafarber Nov 18, 2022
1685000
Merge branch 'matplotlib:main' into issue-23465
kostyafarber Nov 18, 2022
e1ff2e6
Merge branch 'matplotlib:main' into issue-23465
kostyafarber Nov 20, 2022
7d907ed
refactor bar chart examples to use loops
kostyafarber Nov 20, 2022
1a8f253
Update examples/lines_bars_and_markers/bar_label_demo.py
kostyafarber Nov 22, 2022
df93a6f
refactor bar_stacked.py and barchart.py less incomplete data
kostyafarber Nov 24, 2022
e1ddc33
remove if statements from bar_label_demo and bar_stacked
kostyafarber Nov 24, 2022
213b5c7
refactor barchart.py (change xticks and legend) and bar_stacked.py (c…
kostyafarber Nov 29, 2022
3af7b0d
add constrained layout to figure and change ylim setting to OO style
kostyafarber Dec 2, 2022
d78b8a4
Update examples/lines_bars_and_markers/bar_label_demo.py
kostyafarber Dec 6, 2022
2ba5c6a
Update examples/lines_bars_and_markers/bar_stacked.py
kostyafarber Dec 6, 2022
0d11e6e
Update examples/lines_bars_and_markers/barchart.py
kostyafarber Dec 6, 2022
27e2e48
Update examples/lines_bars_and_markers/barchart.py
kostyafarber Dec 6, 2022
255fbfa
Update examples/lines_bars_and_markers/barchart.py
kostyafarber Dec 6, 2022
687c6c8
refactor examples to use dictionaries and fix syntax errors
kostyafarber Dec 6, 2022
1cf307e
Update examples/lines_bars_and_markers/barchart.py
kostyafarber Dec 9, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 14 additions & 22 deletions examples/lines_bars_and_markers/bar_label_demo.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,35 +18,27 @@
import numpy as np

###############################################################################
# Define the data
# data from https://allisonhorst.github.io/palmerpenguins/

N = 5
coffee_means = (20, 25, -10, 32, 10)
tea_means = (30, 13, -14, 21, 17)
coffee_std = (3, 2, 4, 1, 2)
tea_std = (4, 3, 2, 3, 5)
ind = np.arange(N) # the x locations for the groups
width = 0.25 # the width of the bars: can also be len(x) sequence
species = ('Adelie', 'Chinstrap', 'Gentoo')
sexes = ["Male", "Female"]
male = np.array([73, 34, 61])
female = np.array([73, 34, 58])
sex_counts = [male, female]
width = 0.6 # the width of the bars: can also be len(x) sequence

###############################################################################
# Stacked bar plot with error bars

fig, ax = plt.subplots()
bottom = np.zeros(3)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
bottom = np.zeros(3)
bottom = np.zeros(len(species))

if you want to be even more generalt


p1 = ax.bar(ind, coffee_means, width, yerr=coffee_std, label='Coffee')
p2 = ax.bar(ind, tea_means, width,
bottom=coffee_means, yerr=tea_std, label='Tea')
for sex, sex_count in zip(sexes, sex_counts):
p = ax.bar(species, sex_count, width, label=sex, bottom=bottom)
bottom += sex_count

ax.axhline(0, color='grey', linewidth=0.8)
ax.set_ylabel('Scores')
ax.set_title('Scores by group and their beverage choices')
ax.set_xticks(ind, labels=['G1', 'G2', 'G3', 'G4', 'G5'])
ax.legend()
ax.bar_label(p, label_type='center')
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

is white or a gray easier to read?


# Label with label_type 'center' instead of the default 'edge'
ax.bar_label(p1, label_type='center')
ax.bar_label(p2, label_type='center')
ax.bar_label(p2)
ax.set_title('Number of penguins by sex')
ax.legend()

plt.show()

Expand Down
32 changes: 16 additions & 16 deletions examples/lines_bars_and_markers/bar_stacked.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,30 +3,30 @@
Stacked bar chart
=================

This is an example of creating a stacked bar plot with error bars
using `~matplotlib.pyplot.bar`. Note the parameters *yerr* used for
error bars, and *bottom* to stack the coffee's bars on top of the tea's
bars.
This is an example of creating a stacked bar plot
using `~matplotlib.pyplot.bar`.
"""

import matplotlib.pyplot as plt
import numpy as np

# data from https://allisonhorst.github.io/palmerpenguins/

labels = ['G1', 'G2', 'G3', 'G4', 'G5']
tea_means = [20, 35, 30, 35, 27]
coffee_means = [25, 32, 34, 20, 25]
tea_std = [2, 3, 4, 1, 2]
coffee_std = [3, 5, 5, 3, 3]
width = 0.25 # the width of the bars: can also be len(x) sequence
species = ("Adelie", "Gentoo", "Chinstrap")
booleans = ["True", "False"]
above_average_weight = np.array([70, 31, 58])
below_average_weight = np.array([82, 37, 66])
weight_counts = [above_average_weight, below_average_weight]
width = 0.5

fig, ax = plt.subplots()
bottom = np.zeros(3)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

same as above for generalization


ax.bar(labels, tea_means, width, yerr=tea_std, label='Tea')
ax.bar(labels, coffee_means, width, yerr=coffee_std, bottom=tea_means,
label='Coffee')
for boolean, weight_count in zip(booleans, weight_counts):
p = ax.bar(species, weight_count, width, label=boolean, bottom=bottom)
bottom += weight_count

ax.set_ylabel('Scores')
ax.set_title('Scores by group and beverage preferences')
ax.legend()
ax.set_title("Number of penguins with above average body mass")
ax.legend(loc="upper right")

plt.show()
33 changes: 19 additions & 14 deletions examples/lines_bars_and_markers/barchart.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,32 +7,37 @@
bars with labels.
"""

# data from https://allisonhorst.github.io/palmerpenguins/

import matplotlib.pyplot as plt
import numpy as np

species = ("Adelie", "Chinstrap", "Gentoo")
measurements = ['Bill Length', 'Bill Depth']
bill_length = (38.80, 48.83, 47.50)
bill_depth = (18.34, 18.43, 14.98)
penguin_means = [bill_length, bill_depth]

labels = ['G1', 'G2', 'G3', 'G4', 'G5']
tea_means = [20, 34, 31, 35, 27]
coffee_means = [25, 32, 34, 20, 25]

x = np.arange(len(labels)) # the label locations
x = np.arange(len(species)) # the label locations
width = 0.25 # the width of the bars

multiplier = 0
fig, ax = plt.subplots()
rects1 = ax.bar(x - width/2, tea_means, width, label='Tea')
rects2 = ax.bar(x + width/2, coffee_means, width, label='Coffee')

for measurement, penguin_mean in zip(measurements, penguin_means):
offset = (width * multiplier)
rects = ax.bar(x + offset, penguin_mean, width, label=measurement)
ax.bar_label(rects, padding=3)
multiplier += 1

# Add some text for labels, title and custom x-axis tick labels, etc.
ax.set_ylabel('Scores')
ax.set_title('Scores by group and beverage preferences')
ax.set_xticks(x, labels)
ax.set_ylabel('Length (mm)')
ax.set_title('Penguin attributes by species')
ax.set_xticks(x + (width / 2), species)
ax.legend()

ax.bar_label(rects1, padding=3)
ax.bar_label(rects2, padding=3)

fig.tight_layout()

plt.ylim(0, 60)
plt.show()

#############################################################################
Expand Down