from openpyxl import load_workbook
import os

if __name__ == "__main__":
    wb = load_workbook("C:/Users/TEST/Desktop/Data/test.xlsx")  # 작성할 엑셀 파일 불러오기

    ws = wb.active  # 첫번째 Sheet 불러오기
    IDCol = 'A'     # ID 적을 Col
    nameCol = 'B'   # NAME 적을 Col
    emoCol = 'G'    # Col
    angleCol = 'H'  # Col
    ROW = 3         # 시작할 행

    file_list = os.listdir("C:/Users/TEST/Desktop/Image_Folder/") # 엑셀에 적을 Data 경로 적어주기

    for file in file_list:
        if file.count(".") == 1:
            name = file.split('.')[0]
            ID = name.split('_')[0]
            emotion = name.split('_')[3]
            ANGLE = name.split('_')[5]
            print(ID)
            print(name)
            ws[IDCol + str(ROW)] = ID
            ws[nameCol + str(ROW)] = name
            ws[emoCol + str(ROW)] = emotion
            ws[angleCol + str(ROW)] = ANGLE
            ROW += 1

    wb.save("C:/Users/TEST/Desktop/Data/test_result.xlsx")     # 엑셀 저장해주기

+ Recent posts