Início / coding / Perguntas / 78645930
Accepted
AmirX
Asked: 2024-06-20 15:00:59 +0800 CST

Como posso encontrar a primeira linha após várias linhas duplicadas?

  • 772

Meu DataFrame é:

import pandas as pd
df = pd.DataFrame(
    {
        'x': ['a', 'a', 'a','b', 'b','c', 'c', 'c',],
        'y': list(range(8))
    }
)

E este é o resultado esperado. Quero criar uma coluna z:

   x  y    z
0  a  0    NaN
1  a  1    NaN 
2  a  2    NaN
3  b  3    3
4  b  4    NaN
5  c  5    NaN
6  c  6    NaN
7  c  7    NaN

A lógica é:

Quero encontrar a primeira linha após o primeiro grupo de linhas duplicadas. Por exemplo, em column x, o valor aé o primeiro valor duplicado. Quero encontrar uma linha após o atérmino dos valores. E então coloque o ydessa linha por zcoluna.

Esta é a minha tentativa que não me deu o resultado:

m = (df.x.duplicated())
out = df[m]
python

1 respostas

  1. Best Answer

    Uma opção, usando uma máscara personalizada:

    # flag rows after the first group
m = df['x'].ne(df['x'].iat[0]).cummax()

# pick the first one
out = df[m & ~m.shift(fill_value=False)]

    Se o seu primeiro valor for sempre ae você quiser encontrar o primeiro não, avocê também poderá usar:

    m2 = df['x'].eq('a')

out = df[m2.shift(fill_value=False) & ~m2]

    Ou, se você tiver certeza de que há pelo menos uma linha após os as iniciais:

    out = df.loc[[df['x'].ne('a').idxmax()]]

    Saída:

       x  y
3  b  3

    Alguns intermediários (todas as abordagens):

       x  y      m  ~m.shift(fill_value=False)     m2  m2.shift(fill_value=False)  df['x'].ne('a')
0  a  0  False                        True   True                       False            False
1  a  1  False                        True   True                        True            False
2  a  2  False                        True   True                        True            False
3  b  3   True                        True  False                        True             True
4  b  4   True                       False  False                       False             True
5  c  5   True                       False  False                       False             True
6  c  6   True                       False  False                       False             True
7  c  7   True                       False  False                       False             True
    • 1

relate perguntas