Python 常用方法汇总

常用语法

创建一个无序不重复元素集：

set([iterable])
# iterable -- 可迭代对象对象
example： 
>>> set_x =  set('runoob')
>>> set_x
{'r', 'u', 'o', 'b', 'n'}
>>> type(set_x)
<class 'set'>
>>> list_x = list(set_x)
>>> list_x
['r', 'u', 'o', 'b', 'n']
>>> type(list_x)
<class 'list'>

返回重新排序的列表：

sorted([5, 2, 3, 1, 4])

[1, 2, 3, 4, 5] # 默认为升序

遍历获取列表的值和索引

for index, value in enumerate(name):

获取脚本执行路径

import sys
filepath = sys.path[0]
#每次执行脚本时，python会将执行脚本目录加入PYTHONPATH环境变量中(sys.path获取)

根据路径获取路径末端文件名

basename = os.path.basename(file_path)
//去掉后缀
fn = os.path.splitext(filename)[0]
// 获取目录路径
dirpath = os.path.dirname(fpath)

判断文件夹是否存在

if not os.path.exists(path):
      os.makedirs(path)

获取目录指定后缀文件名

dirlist = os.listdir(dirpath)
flist = [item for item in dirlist if os.path.splitext(item)[1] == '.txt']

 当前目录下新建文件夹

from pathlib import Path
# 当前目录下定义路径
DATA_PATH = Path("data")
PATH = DATA_PATH / "mnist"
# parents：如果父目录不存在，是否创建父目录。
# exist_ok：只有在目录不存在时创建目录，目录已存在时不会抛出异常。
PATH.mkdir(parents=True, exist_ok=True)

保留小数位数

a = float(format(a, '.4f'))  #format 返回的是str类型

计算程序运行时间

import time
time_start = time.time()
  程序主代码
  程序主代码
  程序主代码
time_end = time.time()
print('totally cost', time_end - time_start)

datetime格式化输出

import datetime
d = datetime.datetime(2010, 8, 4, 12, 15, 58)
print(f'{d:%Y-%m-%d_%H:%M:%S}')
print(f'{d.strftime("%Y%m%d_%H:%M:%S")}')

 修改文件某行内容

f = open(fpath, 'r+')  # 返回一个文件对象
    last_fp = f.tell()
    line = f.readline()  # 调用文件的 readline()方法
    while line:
        if  re.search(r'\*', line) !=None:
            line = line.replace('*', ' ')
            f.seek(last_fp, 0)
            f.write(line)
            print("%s" % line)
        last_fp = f.tell()
        line = f.readline()
    f.close()