仅在组中删除重复项

2024-10-05 10:35:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我只想从数据帧中删除特定子集中的重复项。在“A”列中的每个“spec”下,我想删除重复项,但我想在整个数据帧中保留重复项(第一个“spec”下可能有一些行与第二个“spec”下的行相同,但在“spec”下,直到下一个“spec”下我想删除重复项)

这是数据帧

测向

  A          B            C
  spec       first        second
  test       text1        text2
  act        text12       text13
  act        text14       text15
  test       text32       text33
  act        text34       text35
  test       text85       text86
  act        text87       text88
  test       text1        text2
  act        text12       text13
  act        text14       text15
  test       text85       text86
  act        text87       text88
  spec       third        fourth
  test       text1        text2
  act        text12       text13
  act        text14       text15
  test       text85       text86
  act        text87       text88
  test       text1        text2
  act        text12       text13
  act        text14       text15
  test       text85       text86
  act        text87       text88

这就是我想要的:

测向

  A          B            C
  spec       first        second
  test       text1        text2
  act        text12       text13
  act        text14       text15
  test       text32       text33
  act        text34       text35
  test       text85       text86
  act        text87       text88
  spec       third        fourth
  test       text1        text2
  act        text12       text13
  act        text14       text15
  test       text85       text86
  act        text87       text88

我可以将数据帧拆分为“小”数据帧,然后在for-loop中删除每个“小”数据帧的副本,最后将它们连接起来,但我想知道是否还有其他解决方案。你知道吗

我也试过,成功了:

dfList = df.index[df["A"] == "spec"].tolist()
dfList = np.asarray(dfList)
for dfL in dfList:
      idx = np.where(dfList == dfL)
      if idx[0][0]!=(len(dfList)-1):
            df.loc[dfList[idx[0][0]]:dfList[idx[0][0]+1]-1]
                     = df.loc[dfList[idx[0][0]]:dfList[idx[0][0]+1]-1].drop_duplicates()
      else:
            df.loc[dfList[idx[0][0]]:] = df.loc[dfList[idx[0][0]]:].drop_duplicates()

编辑: 我必须在结尾加上:

df.dropna(how='all', inplace=True)

但我只是想知道有没有别的解决办法。你知道吗


Tags: 数据testdfactspecidxtext1text2
3条回答

这应该起作用:

df2 = df.drop_duplicates(subset=['A', 'B','C'])

使用groupby+duplicated

df[~df.groupby(df.A.eq('spec').cumsum()).apply(lambda x: x.duplicated()).values]

       A       B       C
0   spec   first  second
1   test   text1   text2
2    act  text12  text13
3    act  text14  text15
4   test  text32  text33
5    act  text34  text35
6   test  text85  text86
7    act  text87  text88
13  spec   third  fourth
14  test   text1   text2
15   act  text12  text13
16   act  text14  text15
17  test  text85  text86
18   act  text87  text88

细节

我们使用cumsum查找特定“spec”条目下的所有行。组标签包括:

df.A.eq('spec').cumsum()

0     1
1     1
2     1
3     1
4     1
5     1
6     1
7     1
8     1
9     1
10    1
11    1
12    1
13    2
14    2
15    2
16    2
17    2
18    2
19    2
20    2
21    2
22    2
23    2
Name: A, dtype: int64

然后对该序列进行分组,并计算每组的重复项:

df.groupby(df.A.eq('spec').cumsum()).apply(lambda x: x.duplicated()).values

array([False, False, False, False, False, False, False, False,  True,
        True,  True,  True,  True, False, False, False, False, False,
       False,  True,  True,  True,  True,  True])

由此,剩下的就是保留那些对应于“False”的行(即,复制)。你知道吗

另一个可能的解决办法是。。。 您可以拥有一个计数器,并使用计数器值从列a创建一个新的列,只要在列值中遇到spec,就增加计数器值。你知道吗

counter = 0
def counter_fun(val):
    if val == 'spec': counter+=1
    return counter

df['new_col'] = df.A.apply(counter_fun)

然后在新列上分组,并删除重复项。你知道吗

相关问题 更多 >

    热门问题