主页 / coding / 问题 / 78645930
Accepted
AmirX
Asked: 2024-06-20 15:00:59 +0800 CST

如何在多条重复行之后找到第一行？

  • 772

我的数据框是：

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

这是预期的输出。我想创建列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

逻辑是：

我想找到第一组重复行之后的第一行。例如在列中x，值a是第一个重复的值。我想在值结束后找到一行a。然后将该y行的作为z列。

这是我的尝试，但没有给出输出：

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

1 个回答

  1. Best Answer

    一个选项是使用自定义掩码：

    # 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)]

    如果您的第一个值始终是a并且您想要找到第一个非，a那么您也可以使用：

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

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

    或者，如果你确定前导 s 后至少有一行a：

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

    输出：

       x  y
3  b  3

    一些中间体（所有方法）：

       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

相关问题