Получить последнюю дату каждого месяца в списке дат в Python

Question

Я использую Python 2.7, PyCharm и Anaconda,

У меня есть list дат, и я хотел бы получить последнюю дату каждого месяц присутствует в массиве.

Есть ли какие-либо функции или библиотеки, которые могли бы мне помочь?

Я прочитал даты из файла CSV и сохранил их как datetime.

У меня есть следующий код:

Dates=[] 
Dates1=[] 
for date in dates: 
    temp=xlrd.xldate_as_tuple(int(date),0) 
    Dates1.append(datetime.datetime(temp[0],temp[1],temp[2])) 

for date in Dates1: 
    if not (date<startDate or date>endDate): 
     Dates.append(date) 

Чтобы сделать это ясно, предположим, у меня есть:

Dates = [2015-01-20, 2015-01-15, 2015-01-17, 2015-02-21, 2015-02-06] 

(Считают быть в формате datetime.)

В списке я d хотел бы получить:

[2015-01-20, 2015-02-21] 

До сих пор я искал googled вокруг, особенно в Stack Overflow, но я мог найти ответы только на то, как я могу получить последнюю дату каждого месяца, но не из указанного пользователем списка.

Answer 1

Pandas может справиться с этой задачей очень хорошо. Загрузите свой CSV в фрейм данных, затем запустите группу по месяцу и найдите максимальную дату, используя функцию агрегата:

import pandas as pd 
import numpy as np 

df = pd.read_csv('/path/to/file/')   # Load a dataframe with your file 
df.index = df['my_date_field']    # set the dataframe index with your date 
dfg = df.groupby(pd.TimeGrouper(freq='M')) # group by month/alternatively use MS for Month Start/referencing the previously created object 

# Finally, find the max date in each month 
dfg.agg({'my_date_field': np.max}) 

# To specifically coerce the results of the groupby to a list: 
dfg.agg({'my_date_field': np.max})['my_date_field'].tolist() 

Answer 2

За год y и месяц m, calendar.monthrange(y, m)[1] возвращает день последнего номера месяца.

Следующий сценарий принимает список datetime объекта под названием dates и создает новый список, month_last_dates, содержащий datetime объекты, соответствующие последней даты каждого месяца, в котором члены dates падения.

import datetime 
import calendar 

tuples = [(2015, 8, 1), (2015, 9, 16), (2015, 10, 4)] 
dates = [datetime.datetime(y, m, d) for y, m, d in tuples] 

month_last_dates = len(dates) * [None] 
for i, date in enumerate(dates): 
    y, m, d = date.year, date.month, date.day 
    last = calendar.monthrange(y, m)[1] 
    print y, m, last # Output for testing purposes. 
    month_last_dates[i] = datetime.datetime(y, m, last) 

Вот эквивалентный скрипт, написанный более сжато с помощью списка понимания:

import datetime 
import calendar 

tuples = [(2015, 8, 1), (2015, 9, 16), (2015, 10, 4)] 
dates = [datetime.datetime(y, m, d) for y, m, d in tuples] 

month_last_dates = [datetime.datetime(date.year, date.month, 
     calendar.monthrange(date.year, date.month)[1]) for date in dates] 

# Output for testing purposes. 
for date in month_last_dates: 
    print date.year, date.month, date.day 

В вашем случае, учитывая список Dates, вы можете создать новый список, как это:

last_dates = [datetime.datetime(date.year, date.month, 
     calendar.monthrange(date.year, date.month)[1]) for date in Dates]