python自动化

一、读写excel文件

1，读excel文件

import xlrd

#打开工作簿
xlsx = xlrd.open_workbook('D:\\BaiduNetdiskDownload\\Python自动化办公\\源码\\01excel\\001_Excel_xlrd读_xlwt写\\7月下旬入库表.xlsx')
#通过索引查找sheet：
sheet0 = xlsx.sheet_by_index(0)
sheet1 = xlsx.sheet_by_index(1)
#通过sheet名查找：
sheet0a = xlsx.sheet_by_name('7月下旬入库表')
#获取单元格内容的三种方法
print(sheet0.cell_value(1, 2))
print(sheet0.cell(1, 2).value)
print(sheet0.row(1)[2].value)
#获取单元格一行或者一列的切片
print(sheet0.row_values(1))
print(sheet0.col_values(1))
#获取sheet数量：
print(xlsx.nsheets)
#获取所有sheet名字：
print(xlsx.sheet_names())
#获取sheet的名称
print(sheet0.name)
#获取sheet的行数
print(sheet0.nrows)
#获取某一行的列数
print(sheet0.row_len(1))

2，写excel文件

import xlwt

#新建工作簿
new_workbook = xlwt.Workbook()
#新建sheet
worksheet = new_workbook.add_sheet('new_test')
#新建单元格，并写入内容
worksheet.write(1, 1, 'test')
#保存
new_workbook.save('D:\\BaiduNetdiskDownload\\Python自动化办公\\源码\\01excel\\001_Excel_xlrd读_xlwt写\\test2.xls')

3，先读后写excel文件

import xlrd
import xlwt

#读取excel文件
xlsx = xlrd.open_workbook('三年二班（各科成绩单）.xls')
#选择指定sheet
sheet = xlsx.sheet_by_index(0)
#依次单元格数据，并统计总分
all_data = []
#统计共有多少学生，集合自动去重
num_set = set()
for row_i in range(1, sheet.nrows):
    num = sheet.cell_value(row_i, 0)
    name = sheet.cell_value(row_i, 1)
    grade = sheet.cell_value(row_i, 3)

    student = {
        'num': num,
        'name': name,
        'grade': grade,
    }
    all_data.append(student)
    num_set.add(num)

#计算总分
sum_list = []
for num in num_set:
    name = ''
    sum = 0
    for student in all_data:
        if num == student['num']:
            sum += student['grade']
            name = student['name']
    sum_stu = {
        'num': num,
        'name': name,
        'sum': sum
    }
    sum_list.append(sum_stu)

#写入新的excel

#新建工作簿
new_workbook = xlwt.Workbook()
#新建sheet
worksheet = new_workbook.add_sheet('2班')
#新建单元格，并写入内容
#写入第一行的表头
worksheet.write(0, 0, '学号')
worksheet.write(0, 1, '姓名')
worksheet.write(0, 2, '总分')
#自动写入后面的内容
for row in range(0,len(sum_list)):
    worksheet.write(row+1,0,sum_list[row]['num'])
    worksheet.write(row+1,1,sum_list[row]['name'])
    worksheet.write(row+1,2,sum_list[row]['sum'])
#保存
new_workbook.save('2班学生总分.xls')

 二、读写文本文件

import os

#查看当前工作目录，修改当前工作目录
os.getcwd()
os.chdir('F:/个人文件夹/写作/python')

#默认'r'读模式，'w'为覆写模式，'a'为追加模式，追加模式自己在前面加换行符\n
f = open('sql.txt')
#read读取整个文件为一个字符串，readline()每次读取一行为一个字符串，readlines()读取为每行的列表
data = f.read()
data1 = f.readlines()
data2 = f.readline()
#用完必须关闭
f.close()

f = open('sql.txt', 'a')
#write()的参数为字符串格式
f.write('\nhuyuan love tianyuan')
f.close()

f = open('sql3.txt', 'w')
f.write(data)
f.writelines(data1)
f.close()