본문 바로가기
● 인공지능, 분석/20.10 NIPA_주가예측

[matplot] 토끼와 거북이가 달리기 시합

by 0ver-grow 2020. 10. 12.
반응형

토끼와 거북이가 달리기 시합을 하기로 했어요.
공정한 경쟁을 위해서 1초마다 토끼와 거북이의 위치를 다른 동물이 기록하기로 하고 경주를 했네요.

그 위치 데이터가 csv파일로 저장되어 있어요. 우리는 csv 파일을 읽어서 토끼와 거북이의 시간별 위치를 그래프로 시각화 해보고자 합니다!

from elice_utils import EliceUtils
from matplotlib import pyplot as plt
import pandas as pd

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

elice_utils = EliceUtils()

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

        fig, ax = plt.subplots()
        ax.plot(df["토끼"], label='토끼')
        ax.plot(df["거북이"], label='거북이')
        ax.legend() # 범례데이터

        # 그래프를 확인하려면 아래 두 줄의 주석을 해제한 후 코드를 실행하세요.
        fig.savefig("plot.png")
        elice_utils.send_image("plot.png")


if __name__ == "__main__":
    main()
반응형