列出今天写的脚本中用到的一些知识点

列出今天写的脚本中用到的一些知识点

1. 拼接路径

os.path.join(path1, path2, ....)

用途：将参数中的所有路径，用当前系统的路径拼接符进行拼接成完成一个路径。

场景：已知某个目录路径，该路径下有很多表格，需要获取这些表格的绝对路径。

root_dir = r'C:/demo' # 已知的目录路径
for excel in os.listdir(root_dir):
    excel_path = os.path.join(root_dir, excel)

2. 获取上一级目录

os.path.dirname(path)

用途：获取path的上一级目录

场景：已知一个文本的绝对路径，想要获取该文本所在的目录路径

file_path = 'C:/demo/file.txt'
file_dir = os.path.dirname(file_path)
# 输出：C:/demo

3. 获取路径中的最后一级文件夹名或文件名

os.path.basename(path)

用途：获取path中最后一级的文件夹名或文件名

场景：已知一个文本的绝对路径，只想要获取该文本的部分，不想要完整的路径

file_path = 'C:/demo/file.txt'
file_dir = os.path.basename(file_path)
# 输出：file.txt

4. 判断某个路径是否存在

os.path.exists(path)

用途：判断path这个路径是否存在，若不存在，则返回False；存在，则返回True。

场景：想判断某个文件是否存在，再来决定是否要删除

file_path = 'C:/demo/file.txt'
if os.path.exists(file_path):
    print('文件存在，需要删除')
else:
    print('文件不存在，不需要删除')

5. 计算组数

场景：有100条数据，按20个为一组，问：共有几组？

total_num = 100
per_group_num = 20

# 写法1
group_num = total_num // per_group_num if total_num % per_group_num == 0 else total_num // per_group_num + 1

# 写法2
group_num = total_num // per_group_num + (total % per_group_num > 0)

# 写法3
import math
group_num = math.ceil(total_num / per_group_num)

解决这个问题的是思路是向上取整，写法1和写法2的区别就在2的代码量更少，而写法3的代码量在三个写法中是最少的，但从性能上来说，导入模块和调用函数的过程肯定是会比直接计算会更慢一点。

6. 获取表格的总行数

如果是使用pandas模块读取的表格，那获取表格的总行数的代码是df.index.stop

df = pd.read_excel(excel_path)
max_row = df.index.stop

7. 获取当前时间

可以用time或datetime来获取

import time
from datetime import datetime

# 使用time模块
cur_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())

# 使用datetime
cur_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

8. 创建文件夹

使用os.mkdir(path)来创建文件夹，但如果文件夹已存在的话，会报FileExistsError错误。所以在创建之前，要先判断文件夹是否已存在。

dst_dir = 'C:/demo/dst_dir'
os.path.exists(dst_dir) or os.mkdir(dst_dir)

9. 递归创建文件夹

如果需要创建多级文件夹，即创建一个文件夹后，需要在该文件夹下再创建子文件夹，那么可以这样写：

dst_dir = 'C:/demo/parent_dir/child_dir'
os.path.exists(dst_dir) or os.makedirs(dst_dir)

10. 复制整个文件夹

如果想复制整个文件夹，包括该文件夹下的所有文件或子文件夹，可以使用shutil.copytree(src_path, dst_path)

src_path = 'C:/demo/src_dir'
dst_path = 'c:/demo2/'

# 将src_dir文件夹复制到demo2文件夹下
import shutil
shutil.copytree(src_path, dst_path)

这里要注意，假设目标文件夹如果已存在，则会出现报错，如果要覆盖已存在的目标文件夹，需要设置参数dirs_exist_ok=True

shutil.copytree(src_path, dst_path, dirs_exist_ok=True)

11. 获取表格某字段的数据，并转为列表

假设想获取demo表中column字段的所有值，并转为列表的形式，代码可以这样写：

import pandas as pd

df = pd.read_excel('demo.xlsx', usecols=['column'])
column_value_list = df['column'].to_list()