Python基础 —— 获取当前文件所在目录，遍历当前目录，获取指定后缀的文件

In [54]:

x

import os, sys

×…In [55]:

# 获取当前目录

os.getcwd()

×Out[55]:

'C:\\Users\\rHotD\\Documents\\GitHub\\Machine_Learning_In_Action\\Machine Learning In Action With Pandas and Scikit-learn\\chapter-04'

…In [56]:

# 获取当前目录的上级目录

os.path.dirname(os.getcwd())

×Out[56]:

'C:\\Users\\rHotD\\Documents\\GitHub\\Machine_Learning_In_Action\\Machine Learning In Action With Pandas and Scikit-learn'

…In [57]:

x

# 列出当前目录下的所有文件

os.listdir(os.getcwd())

×Out[57]:

['.ipynb_checkpoints',
'01-CountVectorizer 和 伯努利贝叶斯测试.ipynb',
'02- 获取当前目录.ipynb',
'email',
'text_content.txt']

…In [58]:

# 获取当前 脚本文件 的名称

os.path.basename(sys.argv[0])

×Out[58]:

'__main__.py'

…In [59]:

dir_list = os.listdir(os.getcwd())

dir_list

×Out[59]:

['.ipynb_checkpoints',
'01-CountVectorizer 和 伯努利贝叶斯测试.ipynb',
'02- 获取当前目录.ipynb',
'email',
'text_content.txt']

…In [60]:

# 获取目录下指定后缀的文件名组成的 list

# -------------

# 方式1：使用正则表达式，姿势好像不是很标准,,.,

import re

pattern = re.compile(r'.*.txt')

xxx = list(map(pattern.findall, dir_list))

[kkk[0] for kkk in xxx if len(kkk)>0]

×Out[60]:

['text_content.txt']

…In [61]:

dir_list

×Out[61]:

['.ipynb_checkpoints',
'01-CountVectorizer 和 伯努利贝叶斯测试.ipynb',
'02- 获取当前目录.ipynb',
'email',
'text_content.txt']

…In [62]:

# -------------

# 方式2：使用 os.path.splittext() 函数

[x for x in dir_list if os.path.splitext(x)[1] == '.txt']

×Out[62]:

['text_content.txt']

…