迭代列的条件以查找其他列值

时间:2019-03-12 09:17:51

标签: python pandas

我正在尝试对“出生”列进行迭代,以找到特定的年龄组来计算这些组的类别项目。

我的数据集如下。

data = pd.DataFrame({'birth': ['1/12/1980', '2/6/1992', '6/6/1975',
'6/3/2000'], 'problem':['cat3', 'cat2', 'cat1' , 'cat1']})


    birth   problem
0   1/12/1980   cat3
1   2/6/1992    cat2
2   6/6/1975    cat1
3   6/3/2000    cat1
..   ......     ....
15488  9/9/2010    cat55

正确的输出可能像这样:

  

Young(1990-2019):Cat1:55,cat2:154,cat3:84,cat6:5554

     

YoungAdult:(1990年至1975年):Cat1:99,cat2:5121,cat3:5445

     

MiddleAged:(1975年之前):cat1:99,cat2:545,cat82:644

删除了标点符号,将类型转换为字符串并尝试了以下操作:

for i in range(len(yas2)):
    yas3.append(int(yas2[i][-4:]))

然后迭代条件,但我无法对问题列进行积分和计数。

1 个答案:

答案 0 :(得分:6)

首先通过to_datetime转换列,通过cut进行分箱,最后通过GroupBy.size获得每组的计数:

data = pd.DataFrame({'birth': ['1/12/1980', '2/6/1992', '6/6/1975','6/3/2000'], 
                     'problem':['cat3', 'cat2', 'cat1' , 'cat1']})

data['birth'] = pd.to_datetime(data['birth'], dayfirst=True)

bins = [0, 1975, 1990, 2019]
labels= ['MiddleAged','YoungAdult','Young']
data['bins'] = pd.cut(data['birth'].dt.year, bins=bins, labels=labels)

print (data)
       birth problem        bins
0 1980-12-01    cat3  YoungAdult
1 1992-06-02    cat2       Young
2 1975-06-06    cat1  MiddleAged
3 2000-03-06    cat1       Young

df = data.groupby(['bins','problem']).size().reset_index(name='count')
print (df)
         bins problem  count
0  MiddleAged    cat1      1
1  YoungAdult    cat3      1
2       Young    cat1      1
3       Young    cat2      1