获得当前文件夹下的文件名存放到列表内,采用列表推导进行遍历。 #获得当前工作的路径 original = os.getcwd()#return a string reprsatanting current working directory os.chdir(original)#change current directory to original # 用列表推导式遍历文件夹下所有文件名放在列表内 filename_no_suffix_list=[ file.split('.')[0] for file in os.listdir() if file.endswith('.xlsm')] # 如果后缀为.xlsm #正常写需要5句,采用上面的只需要1句
filename_no_suffix_list = [] # 存放无后缀的文件名 for file in os.listdir(): # 遍历文件夹下所有文件名放在file内 file_num = len(os.listdir()) # 文件夹内有多少文件数,这里为什么不加文件夹路径? if file.endswith('.xlsm'): # 如果后缀为.xlsm filename_no_suffix_list.append((file.split('.')[0])) ##存放不带带后缀的文件名
上代码: def get_filename_no_suffix(): # original=input('请输入原始数据所在的路径:') original = os.getcwd() os.chdir(original) # 用列表推导式,遍历文件夹下所有文件名放在file内,如列表推导式看不习惯,请用下面注释掉的5句。 filename_no_suffix_list=[ file.split('.')[0] for file in os.listdir() if file.endswith('.xlsm')] # 如果后缀为.xlsm # 遍历文件夹下所有文件名放在file内 # filename_no_suffix_list = [] # 存放无后缀的文件名 # for file in os.listdir(): # 遍历文件夹下所有文件名放在file内 # file_num = len(os.listdir()) # 文件夹内有多少文件数,这里为什么不加文件夹路径? # if file.endswith('.xlsm'): # 如果后缀为.xlsm # filename_no_suffix_list.append((file.split('.')[0])) ##存放不带带后缀的文件名 return (filename_no_suffix_list) # 返回文件夹内的所有文件名。传递到get_xlsm_to_list_by_pandas()函数