将“ group_by”转换为csv格式的熊猫

时间:2019-11-21 17:59:59

标签: python-3.x pandas csv dataframe pandas-groupby

这是我的CSV:-

csv

但是当我尝试使用g.to_csv('path_to_csv')将此df转换为csv格式时,显示的是这样的:-

S_Name      Fruit

  A         Apple
  A         Apple
  A         Apple
  B         Banana
  B         Banana
  C         Chiku

我想要这样的 csv 中的实际输出:-

S_Name      Fruit

  A         Apple
            Apple
            Apple
  B         Banana
            Banana
  C         Chiku

有人可以帮助我吗,我的错误在哪里?

谢谢!

1 个答案:

答案 0 :(得分:1)

我假设您的df已经具有重置索引,但是如果没有,请使用此

g.reset_index(inplace=True)

我们需要做的是填充S_Name列重复出现的空白

我们可以结合使用shift.ne()来发生这些情况,并将索引传递给列表。

idx = g[g['S_Name'].ne(g['S_Name'].shift())].index.tolist()

print(idx)
[0, 2, 4]

然后,我们使用isin来获取它们的逆数,并使用.loc来为您的列空间分配空白值

g.loc[~g.index.isin(idx),'S_Name'] = ' '
print(g)
   S_Name   Fruit
0      A   Apple
1          Apple
2          Apple
3      B  Banana
4         Banana
5      C   Chiku

然后您可以将其输出到csv

g.to_csv('filename.csv',index=False)