python常用模块

114 阅读1分钟

1.Excel操作

import openpyxl

# 创建表,并插入数据
def create_excel(self):
    workbook = openpyxl.Workbook()
    sheet = workbook.active

    sheet.cell(row=1, column=1, value='id')
    sheet.cell(row=1, column=2, value='name')
    sheet.cell(row=1, column=3, value='age')
    sheet.cell(row=1, column=4, value='sex')

    # 保存
    workbook.save(self.excel_source_path)

# 续写Excel表格
def write_to_excel(self, video_name, monitor_name, start_time, end_time):
    # 加载表格
    workbook = openpyxl.load_workbook(self.excel_source_path)
    sheet = workbook.active

    # 计算新写入的行号
    row = sheet.max_row + 1

    # 写入字段值
    sheet.cell(row=row, column=1, value=row - 1)
    sheet.cell(row=row, column=2, value=name)
    sheet.cell(row=row, column=3, value=age)
    sheet.cell(row=row, column=4, value=sex)

    # 保存Excel文件
    workbook.save(self.excel_source_path)

删除一行:sheet.delete_rows(row_id)
获取标题:key = sheet.row_values(0) (第0行数据)
获取某一列:[cell.value for cell in sheet['A']]
修改某行某列数据:sheet.cell(row=row_id, column=6, value=1)

2.OS

判断文件是否存在: os.path.exists(path)
获取目录下所有文件名:os.listdir(path)

3.sys

退出:sys.exit(0)(正常为0,异常为其它数字)