基本カリキュラムテスト-実験

4033 ワード

ウサギとカメの競走の結果
「私と競走しませんか」
ウサギとカメは競走をすることにした.
公平に競争するために、1秒ごとに他の動物がウサギとカメの位置を記録し、競走を行うことにした.
その場所のデータはcsvファイルに保存されます.csvファイルを読み、ウサギとカメの異なる時間の位置をグラフィックで可視化しようとします!

こうやってやってみよう.
race.csvファイルを読み込み、df変数に保存し、インデックスとして時間列を指定します.
plot()関数でウサギと亀の色をグラフィックに設定し、labelを各色の名前に設定します.
凡例の位置を右下にしてください.
from elice_utils import EliceUtils
from matplotlib import pyplot as plt
import pandas as pd

plt.rcParams["font.family"] = 'NanumBarunGothic'

elice_utils = EliceUtils()


# 아래 경로에서 csv파일을 읽어서 시각화 해보세요
# 경로: "./data/race.csv"
df = pd.read_csv("./data/race.csv")
df = df.set_index('시간')

fig, ax = plt.subplots()

# 토끼 컬럼과 거북이 컬럼을 label과 함께 plot()함수로 그려보세요.
ax.plot(
    df.index, df['토끼'], label='토끼',
    color='skyblue'
)
ax.plot(
    df.index, df['거북이'], label='거북이',
    color='orange'
)

# 범례의 위치를 오른쪽 아래로 설정해보세요.
ax.legend(loc='lower right')

# elice에서 그래프 확인하기
fig.savefig("plot.png")
elice_utils.send_image("plot.png")