[ Python library ] 우선주를 보통주로 변환 함수

import numpy as np
import pandas as pd

shcode = '005930' #삼성전자 보통주
shcode_2 = '005935' #삼성전자 우선주

# 우선주인지 체크하여 우선주면 보통주의 종목코드로 변경하여 반환하는 함수 [우선주, 보통주] 세트들을 보강 필요
def get_commonstock(shcode):
    np_preference = np.array([["004365", "004360"], ["005945", "005940"], ["001045", "001040"], ["000815", "000810"], ["00088K", "000880"],["078935", "078930"], ["005935", "005930"]])
    df_preference = pd.DataFrame(data=np_preference, columns=['preference', 'mainstock'])
    for i in range(len(df_preference.index)):
        if df_preference.iloc[i,0] == shcode:
            shcode = df_preference.iloc[i,1]
    return shcode

print(get_commonstock(shcode))
print(get_commonstock(shcode_2))