CODEWITHSUNDEEP
pythonpandasdataframedatetime
EFaden
EFaden

Reputation: 57

How to check if column of datetime values is in a list of dates?

So I have a column with a list of dates in it. I have a list/array with a set of specific dates in it. I want to assign a new column in my dataframe with true/false as to whether or not the specific date was in the list. I have the following, but it doesn't work and I'm not sure why.

__DATELIST = [date(2017, 7, 4), date(2016, 7, 4), ...]

def isholiday(x):
   return x in __DATELIST

df['isholiday'] = df['date'].apply(isholiday)

Any thoughts? The above is always false.

Upvotes: 0

Views: 4124

Answers (2)

cs95
cs95

Reputation: 402333

Convert to datetime using to_datetime, and then use isin to get your mask:

dates = pd.to_datetime([date(2017, 7, 4), date(2016, 7, 4), ...])
df['isholiday'] = df['date'].isin(dates)

Upvotes: 3

Battery_Al
Battery_Al

Reputation: 809

You should try using datetime.datetime objects instead of datetime.date objects to construct your list of dates. Your data types need to be equivalent.

Upvotes: 0

Related Questions