如何让Python脚本无需重复写路径即可读取指定文件夹内的csv、xlsx文件
Python脚本统一指定文件目录解决方案
你可以通过os模块拼接统一目录路径,同时把重复的文件处理逻辑封装为函数,大幅简化代码:
import pandas as pd import os # 此处统一填写所有待读取文件所在的文件夹路径,替换为你本地的实际路径即可 DATA_FOLDER = "./wind_files" def get_wind_dir_mean(file_name): # 自动拼接完整文件路径,无需手动拼接 full_path = os.path.join(DATA_FOLDER, file_name) # 读取xlsx文件,若读取csv只需将下方read_excel替换为read_csv df = pd.read_excel(full_path) # 删除无意义的Unnamed列 df.drop(df.columns[df.columns.str.contains('unnamed', case=False)], axis=1, inplace=True) # 返回风向均值取整结果 return int(df["WindDirection"].mean()) if __name__ == "__main__": print(get_wind_dir_mean("WindDirectionWest2.xlsx")) print(get_wind_dir_mean("WindDirectionSouth2.xlsx")) print(get_wind_dir_mean("WindDirectionEast2.xlsx"))
补充说明
- 后续新增需要读取的文件时,只需要新增一行调用
get_wind_dir_mean(文件名)的代码即可,不需要重复编写路径和数据处理逻辑 - 关于代码块高亮的问题:即使编辑器没有提供Python格式选项,你可以手动在代码块开头的三个反引号后输入
python,即可触发Python语法高亮
内容的提问来源于stack exchange,提问作者Frederik N. Pedersen
相关产品推荐
相关产品推荐

