python工具

目錄

• Python工具
  • 1. argparse
  • 2. tqdm
  • 3. os, shutil
  • 4. csv

Python工具

1. argparse

import argparse

parser = argparse.ArgumentParser(description="descriptions")

parser.add_argument('--angle',type=int,default=0)
parser.add_argument('--choose',action='store_true',defalut=False)

args = parser.parse_args()

2. tqdm

pip install tqdm
from tqdm import *
pBar = tqdm(total=1000)
for i in range(1000):
    pBar.update(1)

詳細信息

3. os, shutil

判斷是不是文件夾：os.path.isdir(path)

判斷是否爲文件： os.path.isfile(path)

判斷文件是否存在： os.path.exists()

分離目錄和文件：dir,file = os.path.split(path) or dir = os.path.dirname(path) or file = os.path.basename(path)

循環中列出路徑： os.listdir(path)

建立文件：touch -> f = open('./test.txt','w')

遞歸移動目錄 move: shutil.move(src,dst)

rename: os.rename(old,new)

建立多級目錄：os.mkdirs(path)

copy: shutil.copy(from,to)

獲取當前命令：os.getcwd()

remove: os.remove(path)

os.remove(path) #刪除文件

os.removedirs(path) #刪除空文件夾

shutil.rmtree(path) #遞歸刪除文件夾

路徑鏈接：os.path.join(a,b,...)

4. csv

reader:

import csv
f = open('./test.csv','r')
reader_csv = csv.reader(f)
reader_csv = list(reader_csv)

for a, b, c in reader_csv:
    print(a,b,c)

f.close()

writer:

import csv
f_o = open('./out.csv','w')

writer = csv.writer(f_o)
writer.writerow([a,b,c,d])

f_o.close()