Python常用模块整理

1
2
3
4
5
6
7
8
9
#coding=utf-8

import random
#生成随机数，浮点类型
a = random.uniform(0, 20)
#控制随机数的精度round(数值，精度)
n=3
print round(a, n)

1
2
3
4
5
6
7
8
9
10
import random
print(random.randint(0,20)) #随机一个大于等于0且小于等于20的整数

print(random.randrange(1,10)) #从指定范围内，按指定基数递增的集合中 获取一个随机数

n = 1
m = 6
print(random.randrange(n,20,m))
#n=奇数，就以6递增奇数,随机选取一个实数；n=偶数同理

1
2
3
4
5
6
7
import random
print(random.choice("Python"))  #随机返回Python中任意一个字母

print(random.choice(["JaingCaiYong", "is", "a", "handsome", "boy"]))   #随机返回参数列表中任意一个元素

print(random.choice(("Tuple", "List", "Dict")))

1
2
3
4
5
6
7
8
import random
# random.sample(sequence, k)
#从指定序列中随机获取指定长度的片断。sample函数不会修改原有序列

list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
res = random.sample(list, 5) # 从list中随机获取5个元素，作为一个片断返回
print(res)

1
2
3
4
5
import random
list = [1,2,3,a,b,c]
random.shuffle(list)
print(list)

1
2
3
4
5
6
7
8
9
10
11
12
import random
def random_num(m):
    code = ''
    for i in range(m):
        ran1 = random.randint(0, 9)
        ran2 = chr(random.randint(65, 90))  #chr() 用一个范围在 range（256）内的（就是0～255）整数作参数，返回一个对应的字符
        add = random.choice([ran1,ran2])
        code = ''.join([code, str(add)])
    return code
rand = random_num(m=6)   #m为要生成的几位数
print(rand)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import os
import re
import shutil
import time
from datetime import date, timedelta

exportfile ='\\ip1\d$\yswt'
targetpath ='\\ip2\d$\GtjaFile\dysrzrq\jzjy\%s' % (date.today() + timedelta(days=-1)).strftime("%Y%m%d")
#% (date.today() + timedelta(days=-1)).strftime("%Y%m%d")  -1表示天数，获取前一交易日的文件，如果想找后n天的，同理days = -n

if not os.path.isdir(targetpath):
    os.makedirs(targetpath)
else:
    pass
file_list = os.listdir(exportfile)  # 获取文件夹下所有的文件
fileall = []
for file in file_list:
    if re.findall('.*%s.*'%time.strftime("%Y%m%d"), file):  # 判断文件是否是今天日期的
        fileall.append(file)
        srcFilename = exportfile + '\' + file  # 找到指定目录下的文件
        shutil.copy(srcFilename, targetpath)
        copyCommand = 'xcopy %s %s  /s /e /c /y /h /r' % (srcFilename, targetpath)  # 拷贝文件到指定目录并覆盖
        if os.system(copyCommand) == 0:  # 执行该copy命令
            print('拷贝当天文件%s个完成' % len(fileall))

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import logging
#第1步：创建日志器对象
#创建日志对象（记录器），默认是warning等级日志
logger = logging.getLogger()

#第2步：创建控制台处理器
console_handler = logging.StreamHandler()
#定义控制台的等级
console_handler.setLevel(level='DEBUG')
#第1步控制器中添加控制台处理器
log.addHandler(console_handler)
logging.debug("这日志是debug级别")
logging.info("这日志是info级别")
logging.warning("这日志是warning级别")
logging.error("这日志是error级别")
logging.critical("这日志是critical级别")

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import logging
#创建日志对象（记录器）
log = logging.getLogger()
#创建文本处理器。filename是日志输出路径，mode文件权限a添加，encoding文件格式
file_handler = logging.FileHandler(filename='d:/log.txt',mode='a',encoding='utf-8')
#定义文本的等级
file_handler.setLevel(level='ERROR')
#第1步控制器中添加控制台处理器
log.addHandler(file_handler)
logging.debug("这日志是debug级别")
logging.info("这日志是info级别")
logging.warning("这日志是warning级别")
logging.error("这日志是error级别")
logging.critical("这日志是critical级别")

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import logging

class log(object):

    def __init__(self, level="DEBUG"):  # 日志处理器
        self.log = logging.getLogger("日志")
        self.log.setLevel(level)

    def get_formatter(self):  #格式器，定义输出的是控制台，还是文本格式
        console_fmt = logging.Formatter(fmt="%(name)s--->%(levelname)s--->%(asctime)s--->%(message)s")
        file_fmt = logging.Formatter(fmt="%(lineno)d--->%(asctime)s---->%(levelname)s--->%(message)s")
        return console_fmt, file_fmt

    def console_handle(self, level="DEBUG"): #控制台处理器
        console_handler = logging.StreamHandler()
        console_handler.setLevel(level)
        #处理器添加格式
        console_handler.setFormatter(self.get_formatter()[0])
        return console_handler

    def file_handle(self,log_path, level="DEBUG"):   #文本处理器
        file_handler = logging.FileHandler(filename=log_path, mode="a", encoding="utf-8")
        file_handler.setLevel(level)
        # 处理器添加格式器
        file_handler.setFormatter(self.get_formatter()[1])
        return file_handler

    def get_log(self,log_path):  #log_path 日志输出的路径地址  如d:/log.txt
        self.log.addHandler(self.console_handle())  # 日志器添加控制台处理器
        self.log.addHandler(self.file_handle(log_path))     # 日志器添加文件处理器
        return self.log

if __name__ == '__main__':
    pass

1
2
total = len(open(filename).readlines())

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import pandas as pd
import csv

class CsvFile:

    def PdWriteCsv(self,writepath):          #使用pandas写入csv
        a = ['第一列行1','第一列行2']
        b = ['第二列行1','第二列行2']
        dataframe = pd.DataFrame({'列1': a ,'列2': b})  #字典中的key值即为csv中列名
        dataframe.to_csv(writepath,index=False,sep=',')   #将DataFrame存储为csv,index表示是否显示行名，default=True

    def ListWriteCsv(self,writepath):            #列表写入csv
        csvFile = open(writepath, 'w', newline='')    # 设置newline,否则两行之间会空一行
        writer = csv.writer(csvFile)
        writer.writerow(["a_name", "b_name"])   # 先写入columns_name列名
        writer.writerows([[1, 2], [2, 3]])  # 写入多行用writerows

    def DictWriteCsv(self,writepath):    # 从字典写入csv文件
        dic = {'one': 1, 'two': 2, 'three': 3}
        csvFile = open(writepath, 'w', newline='')
        writer = csv.writer(csvFile)
        for key in dic:
            writer.writerow([key,dic[key]])

    def ReadCsv(self,writepath):   # 以字典读取csv文件
        csvFile = open(writepath, "r")
        dict_reader = csv.DictReader(csvFile)
        for row in dict_reader:
            print(row,row['a_name'])

if __name__ == '__main__':
    pass
    # CsvFile().ListWriteCsv(writepath=r"D:write.csv")
    # CsvFile().ReadCsv(writepath=r"D:write.csv")