From a60f6ef25be0d1daf957bb1924308f45ebb24019 Mon Sep 17 00:00:00 2001 From: sipp11 Date: Tue, 19 Aug 2014 16:31:53 +0700 Subject: [PATCH] [s7] change match arg --- s7.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/s7.py b/s7.py index 69e3d55..32945c5 100755 --- a/s7.py +++ b/s7.py @@ -12,13 +12,15 @@ import random ''' แก้ตรงนี้ไปก่อนละกัน No of Tree = “4Ts” และ No of R = “5” -No of Tree ของคุณ ==== NOT_ARG --- ถ้าต้องการทั้งหมดใช้ 'all' +No of Tree ของคุณ ==== NoT_MATCH_ARG เป็น list หรือ tuple + - ถ้าจะเอาอันไหนก็ใส่ใน list ไป เช่น เอา 4Ts, 2Ts >> ['4Ts', '2Ts'] + - ถ้าเอาทุกอันจะใส่ ['4Ts', '3Ts', ... ] ให้ครบ หรือใช้ [] ก็ได้ No of R ของคุณ ======= TOTAL -- ถ้าต้องการทั้งหมดก็ = 0 แต่ว่า ผมไม่รับรองว่าจะใช้ได้ทั้งหมดเพราะว่า มันก็ขึ้นกับหลายอย่าง เช่น 1Ts มีไม่กี่อันอาจจะ error มั้ง? โดยเฉพาะเวลา random ''' -NOT_ARG = 'all' -TOTAL = 20 +NoT_MATCH_ARG = ['2Ts', '1Ts'] +TOTAL = 10 SRC_PATH = os.path.join(TARGET_DIR, 'part6') @@ -40,7 +42,7 @@ def get_by_order(): with open(os.path.join(SRC_PATH, '%s-6.csv' % m), 'rb') as f: rows = csv.reader(f) for r in rows: - if NOT_ARG != r[2] and NOT_ARG != 'all': + if NoT_MATCH_ARG and r[2] not in NoT_MATCH_ARG: continue row = [ r[0], # R_id @@ -81,7 +83,7 @@ def get_by_random(): ok = False while not ok: pick = random.choice(lo_list) - if NOT_ARG != pick[2] and NOT_ARG != 'all': + if NoT_MATCH_ARG and pick[2] not in NoT_MATCH_ARG: continue if pick[0] not in picked: picked.append(pick[0])