定义了一个名为 read_xlsx 的函数,从指定的路径(如果提供了路径)读取一个 Excel 文件并将其内容转换为 pandas 的 DataFrame 对象。
# -*- coding: utf-8 -*-
# @Author : author_name
# @Time : 2024/3/20 18:00
import pandas as pd
def read_xlsx(file_name,path=''):
if path:
if path and path[-1] != '/': # path 不为空且最后一个字符不是 "/"
path += '/'
data = pd.read_excel(f'{path}{file_name}')
else:
data = pd.read_excel(f'file_name')
return data
调用方法如下:
if __name__ == '__main__':
path = './'
file_name = 'data.xlsx'
data = read_xlsx(file_name=file_name,path=path)
查看数据的前五条:
print(data.head(5))