pandas中基于列的csv文件分组

kgsdhlau  于 2023-05-05  发布在  其他
关注(0)|答案(1)|浏览(134)

我有一个文件,每一行都有特定的数据

预期输出为

  • 条件一:如果源是'a',则需要使用前缀'seed_'重命名其头文件
  • 条件二:可以使用组的捆绑ID

有什么办法可以从Pandas身上实现吗?

hgb9j2n6

hgb9j2n61#

你可以使用boolean索引来分割你的 Dataframe (group 'a' vs other),然后使用merge

m = df['source'] == 'a'
out = df[m].drop(columns='source').merge(df[~m], on='bundle id', suffixes=('_seed', '_comp'))

输出:

>>> out
  name_seed  bundle id  price_seed  name_comp  price_comp source
0    iphone        123         999  iphone 12         950      b
1    iphone        123         999  iphone 13         975      c
2     apple        345         100    Apple 1          99      c

相关问题