代码之家 › 专栏 › 技术社区 › Sanchit Aluna

如何使用带有日期筛选器的IF-else(SAS-then-do的替代项)创建新列

python-datetime datetime numpy pandas python

Sanchit Aluna · 技术社区 · 7 年前

我想在Python dataFrame中创建一个新列,但我的代码不起作用。

recession = 0
recovery = 0

if data['excl_flg'] == 0 and data['segment'] != 'ignore':
    if (data['vardate'].dt.to_period('M') - '2008-06-30'.dt.to_period('M')) < 0 & (data['vardate'].dt.to_period('M') - '31Mar2010'.dt.to_period('M') >= 0:
         recession = 1
    elif (data['vardate'].dt.to_period('M') - '31Mar2010'.dt.to_period('M')) < 0 and (data['vardate'].dt.to_period('M') - '30Sep2013'.dt.to_period('M')) >= 0:
        recovery = 1

这是我的数据,我正试图创建新的专栏《衰退与复苏》。感谢您的帮助。

1 回复 | 直到 7 年前

Community Mohan Dere 6 年前

试试这个。。不使用if-else,但用于此目的。。

资料

   excl_flag        segment    vardate    obsdate
0          0         CC_620  2/29/2008  3/31/2008
1          1  CC_cur_It_660  2/28/2009  3/31/2009
2          0         ignore  2/28/2010  3/31/2010
3          0         CC_620  2/28/2011  3/31/2011
4          1         ignore  2/29/2012  3/31/2012
5          0         CC_620  2/28/2013  3/31/2013
6          1  CC_cur_It_660  2/28/2014  3/31/2014
7          0         CC_620  2/28/2015  3/31/2015
8          1  CC_cur_It_660  2/29/2016  3/31/2016
9          0         ignore  2/28/2017  3/31/2017

from datetime import datetime
import pandas as pd

将日期转换为日期格式。。

data['vardate'] = pd.to_datetime(data['vardate']).apply(lambda x: x.date())
data['obsdate'] = pd.to_datetime(data['obsdate']).apply(lambda x: x.date())

data['recession'] = 0
data['recovery'] = 0

data[((data['excl_flag'] == 0) & (data['segment'] != 'ignore'))].loc[(data['vardate']- datetime.strptime('2008-06-30',"%Y-%m-%d").date()).apply(lambda x: x.days < 0 )  & ((data['vardate'] - datetime.strptime('31Mar2010',"%d%b%Y").date()).apply(lambda x: x.days >=0)), 'recession'] = 1


data[((data['excl_flag'] == 0) & (data['segment'] != 'ignore'))].loc[(data['vardate']- datetime.strptime('31Mar2010',"%d%b%Y").date()).apply(lambda x: x.days < 0 )  & ((data['vardate'] - datetime.strptime('30Sep2013',"%d%b%Y").date()).apply(lambda x: x.days >=0)), 'recovery'] = 1

print(data)

您的结果为零,因为您指定的条件找不到任何接受者

   excl_flag        segment     vardate     obsdate  recession  recovery
0          0         CC_620  2008-02-29  2008-03-31          0         0
1          1  CC_cur_It_660  2009-02-28  2009-03-31          0         0
2          0         ignore  2010-02-28  2010-03-31          0         0
3          0         CC_620  2011-02-28  2011-03-31          0         0
4          1         ignore  2012-02-29  2012-03-31          0         0
5          0         CC_620  2013-02-28  2013-03-31          0         0
6          1  CC_cur_It_660  2014-02-28  2014-03-31          0         0
7          0         CC_620  2015-02-28  2015-03-31          0         0
8          1  CC_cur_It_660  2016-02-29  2016-03-31          0         0
9          0         ignore  2017-02-28  2017-03-31          0         0