取得所有台股股票代號（一）

import pandas as pd
import os

def getCode(): #確定CSV存在並匯入Dataframe
    if os.path.exists('OriginData/List.csv'):
        codelist = pd.read_csv('OriginData/List.csv',header=0)
        c = codelist['證券代碼']
        n = pd.DataFrame({'Code' : []})
        for i in c:
            i = i.split(" ")
            n = n.append({'Code' : [i[0]]},ignore_index=True)    
        n.to_pickle('Data/codelist.pkl')
        return 1;
    else: #
        print("Data does not exist!")
        return 0;
    
def readCode(): # 讀取Dataframe
    print("Warning: Loading pickled data received from untrusted sources can be unsafe")
    if os.path.exists('Data/codelist.pkl'):
        codelist = pd.read_pickle('Data/codelist.pkl')
        print("Loading...")
        return codelist
    else:
        print("CodeList does not exist, please try again or check")
        getReturnCode = getCode()
        if 1 == getReturnCode:
            print("CodeList retrieved!")
            return readCode()
        else:
            print("Please make sure Data is in OriginData/List.csv")
            return 0;

codelist = readCode()
if isinstance(codelist,pd.DataFrame):
    print(codelist['Code'].head()) #印出Dataframe中第一個股票代號
else:
    print("Error! Please check you have OriginData/List.csv or Data/codelist.pkl")