Python アニメーション ランダムウォーク

Question

ボールが一定速度 ???? で動くアニメーションを生成したいのですが、このままの状態では、ボールは長方形をすり抜けてしまいます。 壁にぶつかったときは跳ね返るようなアニメーションを作成するにはどうしたらいいですか。
途中の一部分がわかりません。

from matplotlib.animation import ArtistAnimation
fig, ax = plt.subplots()
ims = []


W,H=100,300 
b=[0,0] 
v=[11,13]


while len(ims)<100:

  b[0] += v[0]
  b[1] += v[1]


  ax.add_patch(plt.Rectangle(xy=(0,0),width=W,height=H,fill=False))
  im=ax.scatter([b[0]],[b[1]],c="b")
  ims.append([im])

anim = ArtistAnimation(fig, ims, interval=100)
plt.rc('animation', html='jshtml')
plt.close()

anim
コード

Comments

[1T2R3M4]

調べたこと、試したことを質問に追記していただけませんか。

[y_waiwai]

このままではコードが読めないので、質問を編集し、<code>ボタンを押し、出てくる’’’の枠の中にコードを貼り付けてください

Answer 1

「壁にぶつかったときは跳ね返る」ので、速度(のベクトル成分)を反転します。

from matplotlib import pyplot as plt
from matplotlib.animation import ArtistAnimation

W, H, LW = 100, 300, 2
b = [0, 0] 
v = [11, 13]
ims = []

fig, ax = plt.subplots()
ax.add_patch(plt.Rectangle(xy=(-LW, -LW), width=W+2*LW, height=H+2*LW, lw=LW, fill=False))
for _ in range(100):
  prev = b[:]; newv = [1, 1]
  b[0] += v[0]; b[1] += v[1]
  if b[0] > W:
      b = [W, (v[1]/v[0])*(W-prev[0])+prev[1]]
      newv[0] = -1
  if b[0] < 0:
      b = [0, (v[1]/v[0])*-prev[0]+prev[1]]
      newv[0] = -1
  if b[1] > H:
      b = [(v[0]/v[1])*(H-prev[1])+prev[0], H]
      newv[1] = -1
  if b[1] < 0:
      b = [(v[0]/v[1])*-prev[1]+prev[0], 0]
      newv[1] = -1
  v = [v[0]*newv[0], v[1]*newv[1]]
  im = ax.scatter([b[0]], [b[1]], c="b")
  ims.append([im])

anim = ArtistAnimation(fig, ims, interval=100)

plt.rc('animation', html='jshtml')
plt.show()

Comments

[TLimi]

すごく助かりました。ありがとうございました。

[melian]

今回の場合では結果に違いは出ないのですが、範囲チェックの部分を修正しました。