[已解决]模拟天体运动，matplotlib也行

歌者文明清理员

限制：
1 啥都行，就是得计算引力加速度，不能直接画圆
2 像

1. Stars = [Star(radius, mass), Star(radius, mass), …]

复制代码

这样（名字不一定），可以给个示例，我自己改
3 想不到吧，原来的3太离谱所以删了
看之前的帖子冷清了，就再发一次，怕是之前的没人看了

新规矩，链接放后面未知链接
不知道为什么，发不了悬赏

最佳答案

月排行榜 / 总排行榜

鱼cpython学习者

2023-2-10 21:02:49

1. import pygame
   
2. import sys
   
3. import math
   
4. import numpy as np
   
5. import random
   
6. 
   
7. G = 0.3
   
8. 
   
9. 
   
10. class Star:
    
11.     def __init__(self, star_id, radius, mass):
    
12.         self.star_id = star_id
    
13.         self.radius = radius
    
14.         self.mass = mass
    
15.         self.position = [random.randrange(width), random.randrange(height)]  # 随机初始化位置
    
16.         self.speed = [int(random.choice([1, -1]) * 6 * np.random.rand(1)), int(random.choice([1, -1]) * 6 * np.random.rand(1))]  # 随机初始化速度
    
17. 
    
18.     def move(self):
    
19.         self.position[0] += self.speed[0]
    
20.         self.position[1] += self.speed[1]
    
21.         # 跑出窗口从另一边回来
    
22.         self.position[0] -= self.position[0] // width * width
    
23.         self.position[1] -= self.position[1] // height * height
    
24. 
    
25.         pygame.draw.circle(screen, (255, 255, 255), self.position, self.radius)
    
26. 
    
27. 
    
28. if __name__ == '__main__':
    
29.     pygame.init()
    
30. 
    
31.     size = width, height = 1920, 1020
    
32.     bg = (0, 0, 0)
    
33. 
    
34.     screen = pygame.display.set_mode(size)
    
35.     pygame.display.set_caption("天体运动")
    
36. 
    
37.     stars = [Star(star_id, random.randrange(3, 8), random.randrange(100, 200)) for star_id in range(10)]  # 随机生成天体
    
38. 
    
39.     clock = pygame.time.Clock()
    
40. 
    
41.     step = 1
    
42. 
    
43.     while True:
    
44.         clock.tick(60)
    
45.         for event in pygame.event.get():
    
46.             if event.type == pygame.QUIT:
    
47.                 sys.exit(0)
    
48. 
    
49.         screen.fill((0, 0, 0))
    
50.         for i in range(len(stars)):
    
51.             star = stars[i]
    
52.             for star2 in stars[(i + 1):]:
    
53.                 distance = math.dist(star.position, star2.position)
    
54.                 if distance < 100:
    
55.                     distance = 100
    
56.                 # 万有引力公式F=GMm/r**2，0.1是为了适当缩小万有引力，否则会出现天体一直在转圈，你可以自己调整数值
    
57.                 f = (G * (star.mass * star2.mass)) / distance ** 2 * 0.1
    
58.                 relative_position = np.subtract(star.position, star2.position)
    
59.                 angle = math.atan2(*relative_position[::-1])
    
60.                 dx = math.cos(angle) * f
    
61.                 dy = math.sin(angle) * f
    
62.                 star.speed = np.subtract(star.speed, [dx, dy])
    
63.                 star2.speed = np.add(star2.speed, [dx, dy])
    
64.             star.speed = [x - math.copysign(0.001, x) for x in star.speed]  # 虽然天体是在做匀速直线运动，但不做衰减的话速度会一直叠加导致过快
    
65.             star.move()
    
66.         pygame.display.flip()
    

复制代码

初中生，不懂天体运动，找了个github项目作参考
https://github.com/zkytech/Simulation-of-Star-Motion
这个项目是在网页上模拟的，我移植到pygame来，并改了一些重要的地方
有什么bug或者疑惑的请告诉我

跳转到最佳答案楼层

Mike_python小

育碧

Mike_python小

所以这道题是你的作业吗

鱼cpython学习者

你要啥样的运动，是一个天体围绕另一个运动，还是两个天体互相影响的那种

歌者文明清理员

鱼cpython学习者 发表于 2023-2-8 21:55
你要啥样的运动，是一个天体围绕另一个运动，还是两个天体互相影响的那种

互相影响

鱼cpython学习者

1. import pygame
   
2. import sys
   
3. import math
   
4. import numpy as np
   
5. import random
   
6. 
   
7. G = 0.3
   
8. 
   
9. 
   
10. class Star:
    
11.     def __init__(self, star_id, radius, mass):
    
12.         self.star_id = star_id
    
13.         self.radius = radius
    
14.         self.mass = mass
    
15.         self.position = [random.randrange(width), random.randrange(height)]  # 随机初始化位置
    
16.         self.speed = [int(random.choice([1, -1]) * 6 * np.random.rand(1)), int(random.choice([1, -1]) * 6 * np.random.rand(1))]  # 随机初始化速度
    
17. 
    
18.     def move(self):
    
19.         self.position[0] += self.speed[0]
    
20.         self.position[1] += self.speed[1]
    
21.         # 跑出窗口从另一边回来
    
22.         self.position[0] -= self.position[0] // width * width
    
23.         self.position[1] -= self.position[1] // height * height
    
24. 
    
25.         pygame.draw.circle(screen, (255, 255, 255), self.position, self.radius)
    
26. 
    
27. 
    
28. if __name__ == '__main__':
    
29.     pygame.init()
    
30. 
    
31.     size = width, height = 1920, 1020
    
32.     bg = (0, 0, 0)
    
33. 
    
34.     screen = pygame.display.set_mode(size)
    
35.     pygame.display.set_caption("天体运动")
    
36. 
    
37.     stars = [Star(star_id, random.randrange(3, 8), random.randrange(100, 200)) for star_id in range(10)]  # 随机生成天体
    
38. 
    
39.     clock = pygame.time.Clock()
    
40. 
    
41.     step = 1
    
42. 
    
43.     while True:
    
44.         clock.tick(60)
    
45.         for event in pygame.event.get():
    
46.             if event.type == pygame.QUIT:
    
47.                 sys.exit(0)
    
48. 
    
49.         screen.fill((0, 0, 0))
    
50.         for i in range(len(stars)):
    
51.             star = stars[i]
    
52.             for star2 in stars[(i + 1):]:
    
53.                 distance = math.dist(star.position, star2.position)
    
54.                 if distance < 100:
    
55.                     distance = 100
    
56.                 # 万有引力公式F=GMm/r**2，0.1是为了适当缩小万有引力，否则会出现天体一直在转圈，你可以自己调整数值
    
57.                 f = (G * (star.mass * star2.mass)) / distance ** 2 * 0.1
    
58.                 relative_position = np.subtract(star.position, star2.position)
    
59.                 angle = math.atan2(*relative_position[::-1])
    
60.                 dx = math.cos(angle) * f
    
61.                 dy = math.sin(angle) * f
    
62.                 star.speed = np.subtract(star.speed, [dx, dy])
    
63.                 star2.speed = np.add(star2.speed, [dx, dy])
    
64.             star.speed = [x - math.copysign(0.001, x) for x in star.speed]  # 虽然天体是在做匀速直线运动，但不做衰减的话速度会一直叠加导致过快
    
65.             star.move()
    
66.         pygame.display.flip()
    

复制代码

初中生，不懂天体运动，找了个github项目作参考
https://github.com/zkytech/Simulation-of-Star-Motion
这个项目是在网页上模拟的，我移植到pygame来，并改了一些重要的地方
有什么bug或者疑惑的请告诉我

歌者文明清理员

鱼cpython学习者 发表于 2023-2-10 21:02
初中生，不懂天体运动，找了个github项目作参考
https://github.com/zkytech/Simulation-of-Star-Motion ...

谢，可是本小学生（歌者）平时无电脑且上不了github

鱼cpython学习者

歌者文明清理员 发表于 2023-2-10 21:57
谢，可是本小学生（歌者）平时无电脑且上不了github

没事，那个github是如果你要修改一些细节的东西就给你做参考的，你看那个python代码没啥问题就行

歌者文明清理员

鱼cpython学习者 发表于 2023-2-10 22:12
没事，那个github是如果你要修改一些细节的东西就给你做参考的，你看那个python代码没啥问题就行

本人python擅长pygame基础，但是没啥游戏灵感，代码我回复完就看
我是说how to打开github？打开后怎么用？



我之前用turtle的实际数据写过日-地，结果地球总是直线穿过太阳又穿回来然后掉到屏幕下面

鱼cpython学习者

歌者文明清理员 发表于 2023-2-10 22:14
本人python擅长pygame基础，但是没啥游戏灵感，代码我回复完就看
我是说how to打开github？打开后怎么 ...

这个你可以去查，有很多方法，有加速插件，有改hosts之类的，很容易搜到的
例如：
https://blog.csdn.net/weixin_42121805/article/details/107711983
https://blog.csdn.net/nuannuanwfm/article/details/104119932
围绕天体转动和多个天体互相影响运动在物理中好像差别还蛮大的，我的建议是你最好去向一些专业的人寻求帮助，编程论坛一般帮助不大