端到端的机器学习: 001. python os 模块常用函数
python os 模块常用函数
import os
os.getcwd() # 获取当前工作空间/目录
os.mkdir("PASH") # 新建目录
os.chdir("PATH") # 切换新都工作空间
os.listdir() # 列出当前工作空间文件/文件夹
os.path.splitext("PATH") # 分离文件名和扩展名
os.path.join('PATH1', 'PATH1', ...) # 拼接多个目录,
查看指定目录下的所有文件案例1:
import os
path = os.getcwd() # 查看当前工作空间
path_join = os.path.join(path, 'input', '11595')
os.listdir(path_join)
['sample_submission.csv',
'sample_submission.csv.zip',
'test.csv',
'test.csv.zip',
'train.csv',
'train.csv.zip']
查看指定目录下的所有文件案例2:
import os
path_join = os.path.join('input', '11595')
for dirname, _, filenames in os.walk(path_join):
print(dirname)
for filename in filenames:
print(filename)
input\11595
sample_submission.csv
sample_submission.csv.zip
test.csv
test.csv.zip
train.csv
train.csv.zip
查看指定目录下的所有文件案例3:
import os
path_join = os.path.join('input', '11595')
for dirname, _, filenames in os.walk(path_join):
for filename in filenames:
print(os.path.join(dirname, filename))
input\11595\sample_submission.csv
input\11595\sample_submission.csv.zip
input\11595\test.csv
input\11595\test.csv.zip
input\11595\train.csv
input\11595\train.csv.zip
os.path.splitext()
# os.path.splitext()是 Python 的 os 模块中的一个函数,它用来分离文件名和扩展名。
os.path.splitext(path_file)
('input\11595\train.csv', '.zip')
_,
# 其中的 _ 是一个常规的 Python 变量,并且它被用来存储分离的文件名部分。
# 由于文件名部分在这个语句中并不需要,所以采用了 _ 作为变量名,以表示这个值会被忽略。
_, file_type = os.path.splitext(path_file)
file_type
'.zip'