端到端的机器学习:001. python os 模块常用函数

112 阅读1分钟

端到端的机器学习: 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'

参考网址

  1. www.kaggle.com/code/alexis…