[Python] pandas를 이용한 엑셀 xlsx 파일을 df 데이터프레임으로 가져오는 함수(pd.read_excel 사용)

현재경로에서 xlsx 파일을 입력하면, 데이터프레임 형태를 return해주는 python 함수

import pandas as pd
import openpyxl as op
import numpy as np
import os


def pwd():
	#용도: 현재 위치를 알아내는 함수
	#사용법: ef.pwd()
	print(os.getcwd())
	return os.getcwd()
	
def file_pwd(location=''):
	#옹도: 현재경로 + 내 파일경로를 이어주는 함수
	#사용법: ef.file_pwd('data/data_1.xlsx')
	if location=='':
		print("file_pwd 에러: 현재경로에서 지정하려는 파일의 경ㄹ호를 입력해주세요.")
		return 0
	else:		
		#print("파일경로:"+ os.getcwd()+location)
		return(os.getcwd()+location)

def read_xlsx_an_df(location='', sheet='sheet1'):
	#옹도: xlsx파일을 Dataframe형태로 가져오는 파일
	#사용법: ef.read_xlsx_an_df( ef.file_pwd('data/data_1.xlsx') )
	if location =='':
		print("read_xlsx_an_df 함수 에러: 경로를 입력해주세요.")
		return 0
	else:		
		df = pd.read_excel(location, engine='openpyxl')
		wb = op.load_workbook(location)
		#print(wb.sheetnames)
		ws = wb[sheet]
		data = ws.values
		columns = next(data)[0:]
		df = pd.DataFrame(data, columns=columns)
	return df

 

사용코드

import excel_function as ef

df = ef.read_xlsx_an_df(ef.file_pwd(input_file_name))
	print()

 

df.to_excel() : 데이터프레임을 엑셀파일로 저장해준다. 

learning_dataframe이라는 데이터프레임을 learning_data.xlsx 파일로 저장해준다.

learning_dataframe.to_excel('learning_data.xlsx')