python/도구

python으로 csv, excel 다루기

자코린이 2023. 6. 20. 22:00
import pymysql
import csv
import pandas as pd
import openpyxl
from openpyxl import Workbook
from openpyxl import load_workbook




#read data with csv
def open_csv(feature):
    data = list()
    f = open('%s.csv' % feature, 'r', encoding='utf-8')
    rea = csv.reader(f)
    for row in rea:
        data.append(row)
    f.close
    return data

def make_excel(feature):
    wb = Workbook()
    ws = wb.active
    
    # enter your file path
    path = 'excel파일 이름'
 
    # load excel file
    book = openpyxl.load_workbook(path)
    
    
    
    # select the sheet
    sheet = book['sheet 이름']
    

    try:
        #get sheet row count
        maxRow = sheet.max_row
        
    except:
        print('max row fail')
    
    try:
        #delete all row except first row
        sheet.delete_rows(2,maxRow)
        print("delete success")
    except:
        print('delete fail')
    
    try:
        for row in feature:
            #add row in excel
            sheet.append(row)
    except:
        print('insert fail')
    
    try:
        #save excel file
        book.save(path)
        print('save success')
    except:
        print('save fail')

if __name__ == "__main__":

    
    try:
        #change argument to your csv file
        data = open_csv('저장할 이름')
    except:
        print("정보가 없음")
        
    try:
        make_excel(data)
    except:
        print("실행 안됨")

이 프로그램은 csv로 되어있는 정보를 excel로 변환해주는 프로그램입니다.

'python > 도구' 카테고리의 다른 글

가비아 sms 최종 소스  (0) 2022.12.07
가비아 sms 서비스 구축  (0) 2022.12.07
공공데이터포털 데이터를 mysql에 넣기  (0) 2022.08.09
python 에서 원격으로 폴더 생성(paramiko)  (0) 2022.07.22
python 파일 삭제  (0) 2022.07.22