[파이썬] pandas, DataFrame에서 max, min 찾기

 

pandas DataFrame 데이터에서 Max 값을 찾아봅니다.

import pandas as pd

df = pd.DataFrame({'A1': [1, 2, 2, 3],
                   'A2': [4, 3, 8, 4]})
print("DataFrame:")
print(df)

maxs = df.max()

print("\n")
print("Max Column:")
print(maxs)

 

2개 값을 평균값을 사용해서 mean, max, min 값을 찾습니다.

이를 사용해서 주식데이터의 5일 이동평균서의 Open, High, Low 값을 찾을 수 있습니다.

df = pd.DataFrame({'A1': [1, 2, 3, 4, 5, 6, 7, 8, 9],
                   'A2': [11, 12, 13, 14, 15, 16, 17, 18, 19]})

df['d2close'] = df['A1'].rolling(window=2).mean()
df['d2high'] = df['d2close'].rolling(window=2).max()
df['d2low'] = df['d2close'].rolling(window=2).min()
df['d2open'] = df['d2close'].shift(1)

df

 

 

위 코드에서는 연속된 데이터로 확인했습니다.

난수로 발생한 데이터로 다시 한번 확인해 봅니다.

df = pd.DataFrame({'A1': random.sample(range(1, 100), 10),
                   'A2': random.sample(range(1, 100), 10)})

df['d2close'] = df['A1'].rolling(window=2).mean()
df['d2high'] = df['d2close'].rolling(window=2).max()
df['d2low'] = df['d2close'].rolling(window=2).min()
df['d2open'] = df['d2close'].shift(1)

df