[Python] 파이썬으로 주식 종목코드 확인하는 방법

'''
pip install pandas,beautifulsoup4,finance-datareader -y
'''

import FinanceDataReader as fdr

def search_code(name):
    df = fdr.StockListing('KRX')
    code = df[df['Name'] == name]['Code'].to_string(index = False)
    return code

if __name__ == '__main__':
    code_samsung = search_code('삼성전자')
    print(code_samsung, type(code_samsung))

    code_sk = search_code('SK하이닉스')
    print(code_sk, type(code_sk))