pandas로 값을뽑아서 matplot으로 차트 그리기

pandas로 값을뽑아서 matplot으로 차트 그리기

 

이런 우상향 하는 그래프를 그려보겠습니다.

 

프로젝트 구조는 위와 같습니다.

 

 

데이터 파일은 아래와 같습니다.

hello.csv

다운로드

 

 

파일의 내용은 아주 단순합니다.

 

 

소스코드는 아래와 같습니다.

import pandas as pd
import matplotlib.pylab as plt

csv = pd.read_csv("../hello.csv")

arHello = csv.loc[0:2, 'hello']

print(csv)
print(arHello)

plt.plot([1, 2, 3], arHello)
plt.show()

 

 

그런데 위와 같이 짜면 0~2까지만 데이터를 뽑아서 그릴 수 있기 때문에 레코드가 추가 되어도 차트가 바뀌지 않습니다.

import pandas as pd
import matplotlib.pylab as plt

csv = pd.read_csv("hello.csv")

arHello = csv.loc[:, 'hello']
arBye = csv.loc[:, 'bye']

print(csv)
print(type(csv))
print(arHello)

print(type(csv.loc[:, 'hello']))

plt.plot(arBye, arHello)
plt.show()

 

이렇게 0:2 대신 :를 쓰게 바꾸면 데이터 개수를 전체를 가지고 오기 때문에 로우가 업데이트 되어도 차트가 잘 그려집니다.

 

 

계단함수와 그래프

import numpy as np
import matplotlib.pylab as plt
'''
x = y
1   1
2   2
0.5 0.5
'''
x = np.arange(-5, 5, 0.1)
y = lambda x: np.array(x > 2, dtype=int)

print(y(x))

plt.plot(x, y(x))
plt.grid()
plt.show()

 

 

end.