x

import os, sys

…

 

# 获取当前目录

os.getcwd()

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

…

 

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

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

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

…

x

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

os.listdir(os.getcwd())

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

…

 

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

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

'__main__.py'

…

 

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

dir_list

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

…

 

# 获取目录下指定后缀的文件名组成的 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]

['text_content.txt']

…

 

dir_list

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

…

 

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

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

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

['text_content.txt']

…