AI Engineering Topic

인턴 때 배운 코드들

Young_Metal 2022. 11. 23. 15:31

1. xlsx 파일을 불러와서 sheet의 이름을 갖고 그 안의 내용을 모두 Dataframe화하기.

master = openpyxl.load_workbook('파일이름.xlsx')
master.sheetnames
#sheet 이름을 수정하여 사용 가능
sheet = master['mando']
mando = pd.DataFrame(sheet.values)

2. 0행으로 header가 내려올 때, header를 올리는 방법

# 0행으로 header가 내려와서 다시 header로 올립니다.
mando.columns = mando.iloc[0]
mando = mando[1:]

3. parser를 통해서 terminal로 파이썬 코드를 실행하기 위한 작업

    def __init__(self, item_name='basic', concept_name = 'rule'):
        parser = argparse.ArgumentParser()
        parser.add_argument('--filepath','--p',type = str,default = '')
        args = parser.parse_args()
        filepath = args.filepath
        self.master = openpyxl.load_workbook(filepath)
        self.master.sheetnames