public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add option to add ticks on x-axis in BarChart()

Browse source
This commit is contained in:
DangGiaChi 2021-12-16 22:03:29 +07:00
parent b61f1473a5
commit bbf45f95c6
1 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
manimlib/mobject/probability.py
View file

@ -8,6 +8,7 @@ from manimlib.mobject.svg.tex_mobject import TexText
from manimlib.mobject.types.vectorized_mobject import VGroup from manimlib.mobject.types.vectorized_mobject import VGroup
from manimlib.utils.color import color_gradient from manimlib.utils.color import color_gradient
from manimlib.utils.iterables import listify from manimlib.utils.iterables import listify
from numpy import ndarray
EPSILON = 0.0001 EPSILON = 0.0001
@ -149,7 +150,9 @@ class BarChart(VGroup):
"height": 4, "height": 4,
"width": 6, "width": 6,
"n_ticks": 4, "n_ticks": 4,
"include_x_ticks": False,
"tick_width": 0.2, "tick_width": 0.2,
"tick_width_x": 0.15,
"label_y_axis": True, "label_y_axis": True,
"y_axis_label_height": 0.25, "y_axis_label_height": 0.25,
"max_value": 1, "max_value": 1,
@ -165,6 +168,7 @@ class BarChart(VGroup):
if self.max_value is None: if self.max_value is None:
self.max_value = max(values) self.max_value = max(values)
self.n_ticks_x = len(values)
self.add_axes() self.add_axes()
self.add_bars(values) self.add_bars(values)
self.center() self.center()
@ -182,6 +186,17 @@ class BarChart(VGroup):
ticks.add(tick) ticks.add(tick)
y_axis.add(ticks) y_axis.add(ticks)
if self.include_x_ticks == True:
ticks1 = VGroup()
widths = np.linspace(0, self.width, self.n_ticks_x + 1)
label_values = np.linspace(0, len(self.bar_names), self.n_ticks_x + 1)
for x, value in zip(widths, label_values):
tick1 = Line(UP*0.05, DOWN*0.05)
tick1.set_width(self.tick_width_x)
tick1.move_to(x * RIGHT)
ticks.add(tick1)
x_axis.add(ticks1)
self.add(x_axis, y_axis) self.add(x_axis, y_axis)
self.x_axis, self.y_axis = x_axis, y_axis self.x_axis, self.y_axis = x_axis, y_axis
@ -196,7 +211,7 @@ class BarChart(VGroup):
self.add(labels) self.add(labels)
def add_bars(self, values): def add_bars(self, values):
buff = float(self.width) / (2 * len(values) + 1) buff = float(self.width) / (2 * len(values))
bars = VGroup() bars = VGroup()
for i, value in enumerate(values): for i, value in enumerate(values):
bar = Rectangle( bar = Rectangle(
@ -205,7 +220,7 @@ class BarChart(VGroup):
stroke_width=self.bar_stroke_width, stroke_width=self.bar_stroke_width,
fill_opacity=self.bar_fill_opacity, fill_opacity=self.bar_fill_opacity,
) )
bar.move_to((2 * i + 1) * buff * RIGHT, DOWN + LEFT) bar.move_to((2 * i + 0.5) * buff * RIGHT, DOWN + LEFT * 5)
bars.add(bar) bars.add(bar)
bars.set_color_by_gradient(*self.bar_colors) bars.set_color_by_gradient(*self.bar_colors)