opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-30 05:24:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

After running 2to3

Browse Source
This commit is contained in:
Grant Sanderson
2018-08-09 17:56:05 -07:00
parent 06a65190e7
commit 858051a806
172 changed files with 2117 additions and 2221 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
active_projects/eop/bayes.py
View File

@ -32,9 +32,9 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
def add_cards(self):
you, her = self.you, self.her
community_cards = VGroup(*map(
community_cards = VGroup(*list(map(
PlayingCard, self.community_card_values
))
)))
community_cards.arrange_submobjects(RIGHT)
community_cards.move_to(self.community_cards_center)
deck = VGroup(*[
@ -47,7 +47,7 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
you.hand = self.get_hand(you, self.your_hand_values)
her.hand = self.get_hand(her, None)
hand_cards = VGroup(*it.chain(*zip(you.hand, her.hand)))
hand_cards = VGroup(*it.chain(*list(zip(you.hand, her.hand))))
self.add(deck)
for group in hand_cards, community_cards:
@ -70,10 +70,7 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
for card in you.hand.target:
card.set_height(community_cards.get_height())
selected_community_cards = VGroup(*filter(
lambda card : card.numerical_value >= 10,
community_cards
))
selected_community_cards = VGroup(*[card for card in community_cards if card.numerical_value >= 10])
card_cmp = lambda c1, c2 : cmp(
c1.numerical_value, c2.numerical_value
)
@ -118,10 +115,7 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
def show_flush_potential(self):
you, her = self.you, self.her
heart_cards = VGroup(*filter(
lambda c : c.suit == "hearts",
self.community_cards
))
heart_cards = VGroup(*[c for c in self.community_cards if c.suit == "hearts"])
heart_cards.save_state()
her.hand.save_state()
@ -165,7 +159,7 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
("hearts", "hearts"),
]
for new_suit_pair in new_suit_pairs:
new_symbols = VGroup(*map(SuitSymbol, new_suit_pair))
new_symbols = VGroup(*list(map(SuitSymbol, new_suit_pair)))
for new_symbol, heart in zip(new_symbols, hearts):
new_symbol.replace(heart, dim_to_match = 1)
self.play(Transform(
@ -229,11 +223,11 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
)
self.wait(3)
equation.remove(percentage)
self.play(*map(FadeOut, [
self.play(*list(map(FadeOut, [
equation,
num_hearts, num_hearts_arrow,
num_cards, num_cards_arrow,
]))
])))
self.percentage = percentage
@ -259,9 +253,9 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
FadeIn(sample_space),
ReplacementTransform(percentage, top_label)
)
self.play(*map(GrowFromCenter, [
self.play(*list(map(GrowFromCenter, [
brace for brace in braces
]))
])))
self.wait(2)
self.play(Write(bottom_label))
self.wait(2)
@ -289,7 +283,7 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
self.play(hand.restore)
self.wait()
self.wait()
self.play(*map(FadeOut, it.chain(*hand_lists)))
self.play(*list(map(FadeOut, it.chain(*hand_lists))))
def place_high_bet(self):
you, her = self.you, self.her
@ -415,7 +409,7 @@ class IntroducePokerHand(PiCreatureScene, SampleSpaceScene):
def get_hand(self, pi_creature, keys = None):
if keys is not None:
hand = VGroup(*map(PlayingCard, keys))
hand = VGroup(*list(map(PlayingCard, keys)))
else:
hand = VGroup(*[
PlayingCard(turned_over = True)
@ -585,7 +579,7 @@ class UpdatePokerPrior(SampleSpaceScene):
self.wait()
self.play(Write(VGroup(*label[-2:])))
self.wait(2)
self.play(*map(FadeOut, [her, her.glasses]))
self.play(*list(map(FadeOut, [her, her.glasses])))
self.sample_space.add(brace, label)
self.bottom_explanation = explanation
@ -619,13 +613,13 @@ class UpdatePokerPrior(SampleSpaceScene):
for num in numbers
])
questions = VGroup(*map(TextMobject, [
questions = VGroup(*list(map(TextMobject, [
"Does she bluff?",
"How much does she have?",
"Does she take risks?",
"What's her model of me?",
"\\vdots"
]))
])))
questions.arrange_submobjects(DOWN, aligned_edge = LEFT)
questions[-1].next_to(questions[-2], DOWN)
questions.scale(0.7)
@ -637,13 +631,13 @@ class UpdatePokerPrior(SampleSpaceScene):
ShowCreation(bubble),
Write(words, run_time = 2)
)
self.play(*map(ShowCreation, arrows))
self.play(*list(map(ShowCreation, arrows)))
self.play(numbers.set_color, YELLOW)
self.play(Blink(randy))
self.play(randy.change_mode, "maybe")
self.play(*map(FadeOut, [
self.play(*list(map(FadeOut, [
bubble, words, arrows
]))
])))
for question in questions:
self.play(
FadeIn(question),
@ -679,7 +673,7 @@ class UpdatePokerPrior(SampleSpaceScene):
(0.97, 0.3),
]
self.play(*map(ShowCreation, rects))
self.play(*list(map(ShowCreation, rects)))
self.play(FadeOut(rects))
for i, value in enumerate(it.chain(*new_conditionals)):
self.play(
@ -718,12 +712,12 @@ class UpdatePokerPrior(SampleSpaceScene):
self.play(MoveToTarget(low_bet_space))
self.play(
Write(words),
*map(ShowCreation, arrows)
*list(map(ShowCreation, arrows))
)
self.wait()
for rect in high_bet_space:
self.play(Indicate(rect, scale_factor = 1))
self.play(*map(FadeOut, [words, arrows]))
self.play(*list(map(FadeOut, [words, arrows])))
self.high_bet_space = high_bet_space
@ -758,11 +752,11 @@ class UpdatePokerPrior(SampleSpaceScene):
self.play(MoveToTarget(rects[0]))
self.wait()
self.play(*it.chain(
map(Write, [frac_line, plus]),
map(MoveToTarget, rects[1:])
list(map(Write, [frac_line, plus])),
list(map(MoveToTarget, rects[1:]))
))
self.wait(3)
self.play(*map(FadeOut, [arrow, fraction] + rects))
self.play(*list(map(FadeOut, [arrow, fraction] + rects)))
self.posterior_tex = posterior_tex
@ -821,7 +815,7 @@ class UpdatePokerPrior(SampleSpaceScene):
self.wait()
self.play(post_words[1].fade, 0.8)
self.wait(2)
self.play(*map(FadeOut, [post_words, post_arrow]))
self.play(*list(map(FadeOut, [post_words, post_arrow])))
def preview_tweaks(self):
post_rects = self.post_rects
@ -830,7 +824,7 @@ class UpdatePokerPrior(SampleSpaceScene):
(0.97, 0.3, 1./22),
]
for new_values in new_value_lists:
for i, value in zip(range(2), new_values):
for i, value in zip(list(range(2)), new_values):
self.play(*self.get_conditional_change_anims(
i, value, post_rects
))
@ -863,7 +857,7 @@ class UpdatePokerPrior(SampleSpaceScene):
arrows.next_to(prior_rects[1], RIGHT, SMALL_BUFF)
self.wait(2)
self.play(*map(FadeIn, [her, her.glasses]))
self.play(*list(map(FadeIn, [her, her.glasses])))
self.play(LaggedStart(FadeIn, risk_averse_words))
self.play(her.change_mode, "sad", Animation(her.glasses))
self.wait()
@ -920,7 +914,7 @@ class UpdatePokerPrior(SampleSpaceScene):
*self.get_prior_change_anims(1./22, post_rects),
run_time = 2
)
self.play(*map(FadeOut, [her, her.glasses]))
self.play(*list(map(FadeOut, [her, her.glasses])))
def compute_posterior(self):
prior_rects = self.get_prior_rectangles()
@ -978,7 +972,7 @@ class UpdatePokerPrior(SampleSpaceScene):
])
self.play(Write(fraction.get_part_by_tex("over")))
for pair in zip(pre_top_rect_products, products[0]):
self.play(*map(Indicate, pair))
self.play(*list(map(Indicate, pair)))
self.wait()
self.wait()
self.play(Indicate(prior_rects[1], scale_factor = 1))
@ -993,7 +987,7 @@ class UpdatePokerPrior(SampleSpaceScene):
])
self.wait()
for pair in zip(pre_bottom_rect_products, products[1]):
self.play(*map(Indicate, pair))
self.play(*list(map(Indicate, pair)))
self.wait()
self.play(
Write(fraction.get_part_by_tex("+")),
@ -1213,10 +1207,7 @@ class BayesianNetworkPreview(Scene):
node.ghost = node.fill.copy().fade()
self.add(node.ghost)
all_ghosts.add(node.ghost)
connected_nodes = filter(
lambda n : n not in covered_nodes,
it.chain(node.children, node.parents)
)
connected_nodes = [n for n in it.chain(node.children, node.parents) if n not in covered_nodes]
for next_node in connected_nodes:
if next_node in covered_nodes:
continue
@ -1454,7 +1445,7 @@ class GeneralizeBayesRule(SampleSpaceScene):
self.play(
self.bayes_rule_words.to_edge, UP,
Write(name),
*map(ShowCreation, arrows+rects)
*list(map(ShowCreation, arrows+rects))
)
self.wait()
@ -1517,9 +1508,9 @@ class GeneralizeBayesRule(SampleSpaceScene):
prior_rects.copy(), post_rects,
run_time = 2
))
self.play(*map(FadeIn, [
self.play(*list(map(FadeIn, [
post_rects.braces, post_rects.labels
]))
])))
self.wait()
self.play(*self.get_conditional_change_anims(1, 0.2, post_rects))
self.play(*self.get_conditional_change_anims(0, 0.6, post_rects))
@ -1609,15 +1600,15 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
labels.scale(0.7)
braces, labels = sample_space.get_side_braces_and_labels(labels)
VGroup(sample_space, braces, labels).to_edge(LEFT)
words = map(TextMobject, [
words = list(map(TextMobject, [
"Blunt honesty", "Some confidence"
])
]))
for word, part in zip(words, sample_space.horizontal_parts):
word.scale(0.6)
word.move_to(part)
self.play(LaggedStart(FadeIn, sample_space, run_time = 1))
self.play(*map(GrowFromCenter, braces))
self.play(*list(map(GrowFromCenter, braces)))
for label in labels:
self.play(Write(label, run_time = 2))
self.wait()
@ -1626,7 +1617,7 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
self.change_pi_creature_with_guitar(mode)
self.wait()
self.wait()
self.play(*map(FadeOut, words))
self.play(*list(map(FadeOut, words)))
self.sample_space = sample_space
@ -1643,7 +1634,7 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
for friend in friends:
friend.look_at(randy.eyes)
headphones = VGroup(*map(Headphones, friends))
headphones = VGroup(*list(map(Headphones, friends)))
self.play(FadeIn(friends))
self.pi_creatures.add(*friends)
@ -1701,7 +1692,7 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
),
)
self.wait(2)
self.play(*map(FadeOut, [bubble, content]))
self.play(*list(map(FadeOut, [bubble, content])))
def friends_dont_like(self):
friends = self.friends
@ -1737,7 +1728,7 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
)
self.change_pi_creature_with_guitar("happy")
self.wait()
self.play(*map(FadeOut, [bubble, content]))
self.play(*list(map(FadeOut, [bubble, content])))
self.bubble = bubble
@ -1923,10 +1914,10 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
run_time = 2,
))
self.wait(time)
self.play(*map(FadeOut, [
self.play(*list(map(FadeOut, [
prior_rect, post_rect,
self.ratio_group, self.post_rhs
]))
])))
self.prior_num_rect = prior_rect
@ -1943,11 +1934,11 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
self.wait(2)
def fade_out_post_rect(self):
self.play(*map(FadeOut, [
self.play(*list(map(FadeOut, [
self.post_rects,
self.post_rects.braces,
self.post_rects.labels,
]))
])))
self.play(self.negative_space.restore)
def get_negative_feedback(self):
@ -2010,7 +2001,7 @@ class MusicExample(SampleSpaceScene, PiCreatureScene):
content.scale(0.6)
bubble.add_content(content)
self.play(*map(MoveToTarget, friends))
self.play(*list(map(MoveToTarget, friends)))
self.play(
ShowCreation(bubble),
Write(bubble.content)
@ -2148,7 +2139,7 @@ class FinalWordsOnRule(SampleSpaceScene):
(0.97, 0.3, 1./22),
]
for new_values in new_value_lists:
for i, value in zip(range(2), new_values):
for i, value in zip(list(range(2)), new_values):
self.play(*self.get_conditional_change_anims(
i, value, post_rects
))