Pandas：标记组，然后更改数据结构

ttcibm8c 于 2023-01-11 发布在其他

关注(0)|答案(2)|浏览(154)

以下是我的原始数据：

raw_data =  pd.DataFrame({'Year': [1991, 1991, 1991, 2000, 2000],
                          'ID': ['A', 'A', 'A', 'B', 'B',],
                          'Group': ['a', 'b', 'c', 'a', 'b'],
                          'score': [6252, 6252,6252, 2342, 2342]})

我需要生成三个组列，指示每个ID是否属于该组。透视功能只能改变数据结构，并实现了我的部分目标。

out_data = pd.DataFrame({'Year': [1991, 2000],
             'Group a':['Yes','Yes'],
             'Group b':['Yes','Yes'],
             'Group c':['Yes','No'],
             'ID': ['A', 'B'],
             'score': [6252, 2342]})

pandas

来源：https://stackoverflow.com/questions/73109943/pandas-flag-groups-and-then-change-the-data-structure

2条答案

按热度按时间

n7taea2i1#

这是pivot_table的一个变体：

(df
 .pivot_table(index=['Year', 'ID'], columns='Group', values='score', aggfunc=any)
 .replace({True: 'Yes'}).fillna('No')
 .add_prefix('Group_')
 .reset_index().rename_axis(columns=None)
)

或crosstab：

(pd
 .crosstab([df['Year'], df['ID']], df['Group'], values=df['score'], aggfunc=any)
 .replace({True: 'Yes'}).fillna('No')
 .add_prefix('Group_')
 .reset_index().rename_axis(columns=None)
)

输出：

Year ID Group_a Group_b Group_c
0  1991  A     Yes     Yes     Yes
1  2000  B     Yes     Yes      No

赞(0）回复(0）举报 2023-01-11

utugiqy62#

def function1(dd:pd.DataFrame):
    return dd.assign(col1=1).pivot_table(index=['Year','ID','score'],columns='Group',values='col1')\
        .add_prefix('Group ')

raw_data.groupby(['Year','ID']).apply(function1)\
    .applymap(lambda x:"Yes" if pd.notna(x) else 'No')\
    .droplevel([0,1]).reset_index()

输出：

Group  Year ID  score Group a Group b Group c
0      1991  A   6252     Yes     Yes     Yes
1      2000  B   2342     Yes     Yes      No

赞(0）回复(0）举报 2023-01-11

我来回答

Pandas：标记组，然后更改数据结构

2条答案

相关问题

热门标签

最新问答