2016-07-04 58 views
0

說我有在以下五種方式之一格式化日期列表:更多Effecient方式格式化日期在Python中的MySQL

Date_Type_1 = 2001 Apr 15 
Date_Type_2 = 2001 Apr 
Date_Type_3 = 2000 Spring 
Date_Type_4 = 2000 Nov-Dec 
Date_Type_5 = 2001 

所以,我的最新名單看起來是這樣。

Date_list = ["2001 Apr 15", "2004 May 15", "2011 Jan", "2011 Fall", "2000 Nov-Dec", "2012", "2000 Spring" ] 

我現在想嘗試格式化這些日期,以便它們可以傳遞給MySQL數據庫。我知道如何使用大量的條件流程。這是我如何做到這一點的一個例子。我不會包含所有條件,因爲它會佔用太多空間。

for i in Date_list: 
    year = i[:4] 
    #This will be my conditional logic to define a month. 
    #This is only an example. Will not include all 12 months, 4 seasons, and various bi-months 
    if "Apr" in i and "Mar-Apr" not in i: 
    month = 4 
    if "Mar-Apr" in i: 
    month = 3 
    if "May" in i and "May-Jun" not in i: 
    month = 5 
    if "Apr-May" in i: 
    month = 4 
    if "Spring" in i: 
    month = 3 
    #This will be conditional logic to define the day. 
    #I would do this for each of the 31 days of the month. 
    if "15" in i and "2015" not in i: 
    day = 15 

date_return = datetime.datetime(year,month,day) 
date_format = date_return.date().isoformat 

問題在於我正在做出一些假設。我可以將季節定義爲「春夏」,雙月(例如三月/四月)作爲特定月份返回。這個問題,至少在定義天數的時候是不會趕上日子的:

test_list = [2011 May, 2015 Apr 15] 
for i in test_list: 
    if "15" in i and "2015" not in i: 
    day = 15 

這不會趕上一天。我想知道是否有更有效的方法來做到這一點?目前的這種方法需要50個條件語句來定義日/月。

+0

我不認爲你的代碼很慢,但看起來像你的高效率意味着更短的版本。 – YOU

+1

這兩個都是很好的答案,非常優雅。謝謝,我希望我可以選擇兩者。 – Sean

回答

1

你應該使用Python的正則表達式模塊re。這比嘗試去混合切片和in好很多。

import re 

MONTHS = [ 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 
    'Aug', 'Sep', 'Oct', 'Nov', 'Dec'] 
SEASONS = { 'Spring': 'Mar', 'Summer': 'Jun', 'Fall': 'Sep', 
    'Autumn': 'Sep', 'Winter': 'Dec' } 

_MONTHS_RE = '|'.join(MONTHS) 
_SEASONS_RE = '|'.join(SEASONS) 
DATE_RE = re.compile(r"""(?ax) # ASCII-only verbose mode 
    (?P<year>20[0-9]{2}) # Year 
    (# followed by either... 
     (?P<month>""" + _MONTHS_RE + r""") # a month name then... 
     ( -(?P<endmonth>""" + _MONTHS_RE + r""") # a month range 
     | (?P<day>[1-9][0-9]?) # a day number 
     )? # range and day are optional 
    | (?P<season>""" + '|'.join(SEASONS) + r""") # or a season. 
    )""") 

def parse_date(datestr): 
    m = DATE_RE.match(datestr) 
    if m is None: 
     return # Didn't match 
    md = m.groupdict() 
    year = int(md["year"]) 
    if "season" in md: 
     month = SEASONS[md["season"]] 
    else: 
     month = md["month"] 
     if "endmonth" in md: 
      # handle month range here. 
     day = int(md.get("month", 1)) 
    return year, month, day 
    # Month is returned as a string; to get a number, use: 
    return year, MONTHS.index(month) + 1, day 

請注意,這並不能確保日期存在;它會接受"2099 Jun 50"或許多其他不好的日期。但我會留下過濾,作爲讀者的練習。

1

我認爲你可以做這樣的:

>>> import datetime 
>>> dates = ["2001 Apr 15", "2004 May 15", "2011 Jan", "2011 Fall", "2000 Nov-Dec", "2012", "2000 Spring" ] 
>>> 
>>> def convert(date_str): 
...  tokens = date_str.split(' ') 
...  if len(tokens) == 1: 
...   date_time = datetime.datetime.strptime(date_str, '%Y') 
...  elif len(tokens) == 3: 
...   date_time = datetime.datetime.strptime(date_str, '%Y %b %d') 
...  elif len(tokens) == 2 and '-' in tokens[1]: 
...   date_str = date_str.split('-')[0] 
...   date_time = datetime.datetime.strptime(date_str, '%Y %b') 
...  else: 
...   seasons = { 
...    'spring': 'Mar', 
...    'fall': 'Sep', 
...   } 
...   if tokens[1].lower() in seasons.keys(): 
...    date_str = '{} {}'.format(tokens[0], seasons[tokens[1].lower()]) 
...   date_time = datetime.datetime.strptime(date_str, '%Y %b') 
...  return date_time.date().isoformat() 
... 
>>> 
>>> for date_str in dates: 
...  print '{} === {}'.format(date_str, convert(date_str)) 
... 
2001 Apr 15 === 2001-04-15 
2004 May 15 === 2004-05-15 
2011 Jan === 2011-01-01 
2011 Fall === 2011-09-01 
2000 Nov-Dec === 2000-11-01 
2012 === 2012-01-01 
2000 Spring === 2000-03-01