Python 常用汇总

Python 常用文件操作总结：

from random import shuffle

判断文件夹是否有效：
if not os.path.isdir(targetDir): 
    print ‘Invalid target directory: {}‘.format(targetDir)
    sys.exit(2)  
判断文件夹是否存在：    
if not os.path.exists(targetDir):    
    os.makedirs(targetDir)
    
列出文件夹下的所有文件的名字，不包括路径：
for file in os.listdir(sourceDir):

循环获得文件目录结构下的各个文件：
http://www.cnblogs.com/herbert/archive/2013/01/07/2848892.html
for dir_info in os.walk(image_dir):  
    root_dir, sub_dirs, file_names = dir_info  
    for each in dir_info[2]:
        xmlName = each.replace(‘.jpg‘, ‘.xml‘) #如果目录下都是jpg文件，则将其名字提取，后缀替换为.xml,然后赋值给XMLName，当然，原来的each 名字不变，

if os.path.isfile(in_image = os.path.join(root_dir, file_names)):  #文件拼接，获得全路径，并判断文件是否存在
    print ‘the file with full path is‘, in_image 
if not os.path.exists(targetFile): #判断文件是否存在
    if (os.path.getsize(targetFile) != os.path.getsize(sourceFile))):
        open(targetFile, "wb").write(open(sourceFile, "rb").read()) #文件读写：打开源文件，写入目标文件

os.remove(targetFile) #文件删除
shutil.copy(sourceDir,  targetDir) #复制源文件到指定目录，或者可以复制源文件到指定目录


文件读写
file=open(‘labels.txt‘,‘r‘)
for eachline in file:
    filename1 = eachline.strip().split(‘,‘)  #每一行文件格式：785,533,905,644,14794983.jpg,Car
    filename2 = filename1.replace(‘.jpg‘, ‘.xml‘) 
    newline = ‘dir_path‘ + ‘/‘ + filename2
    txt = open(‘xml.txt‘,‘a‘)
    txt.writelines(newline)
    txt.write(‘\n‘)
    txt.close()
file.close()

有序字典：
from collections import OrderedDict
typenames = OrderedDict()
typenames[‘name1‘] = [0, 0]
typenames[‘name2‘] = [1, 2]
所以有typenames[3] 为[1, 2] typenames[3][0] 为1， typnames[3][1] 值为2

对输入的处理：
import argparse
def get_parse_args():
    parser = argparse.ArgumentParser(description=‘get the args‘)
    parser.add_argument(‘--device‘, dest=‘device_type‘, help=‘device to use‘, default=‘cpu‘, type=str)
    if len(sys.argv) == 1:    
        parser.print_help() #直接可以调用对应的help输出对应的描述  
        sys.exit(1)           
                          
    args = parser.parse_args()
    return args  
    
# 使用的时候    
if __name__ == ‘__main__‘:
    args = parse_args()
    print(args)
    if args.device_type is not None:
        dosomething(args.device_type)