Coding/Python 삽질기

[Python] Cycloid 곡선 그리기

chobocho 2023. 3. 9. 01:03

문명과 수학을 읽는 중 문득 Cycloid 곡선을 그리는 코드가 궁금해서, ChatGPT에게 물어보았다

Cycloid 곡선

 

import numpy as np
import matplotlib.pyplot as plt

# 파라미터 정의
r = 1  # 원의 반지름
a = 1  # 원주와 돌기의 간격

# 각도 범위 설정
t = np.linspace(0, 4*np.pi, 1000)

# 싸이클로이드의 수식
x = r * (t - np.sin(t))
y = r * (1 - np.cos(t)) - a

# 그래프 그리기
fig, ax = plt.subplots(figsize=(8, 8))

ax.plot(x, y, color='blue', linewidth=2)
ax.set_aspect('equal', adjustable='box')
ax.set_title('Cycloid')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.grid(True)

plt.show()

 

위 코드를 실행하면 아래 그림 같이 이쁘게 코드를 그려준다.