主页 / coding / 问题 / 78168551
Accepted
x_Amir_x
Asked: 2024-03-16 00:57:06 +0800 CST

如果掩码的第一个实例位于前 N 行，如何获取它？

  • 772

这是我的数据框。

import pandas as pd
df = pd.DataFrame(
    {
        'a': [100, 1123, 9999, 100, 1, 954, 1],
        'b': [1000, 11123, 1123, 0, 55, 0, 1],
    },
)

预期输出正在创建列c：

      a      b       c
0   100   1000     NaN
1  1123  11123     NaN
2  9999   1123  9999.0
3   100      0     NaN
4     1     55     NaN
5   954      0     NaN
6     1      1     NaN

面具是：

mask = ((df.a > df.b))

我想获得满足此掩码条件的第一行（如果它位于前 3 行）并放入df.acreate c。对于此示例，此代码有效：

df.loc[mask.cumsum().eq(1) & mask, 'c'] = df.a

但对于此 DataFrame，它应该返回NaNfor c，因为 的第一个实例mask不在前 3 行中。但这不起作用。

df = pd.DataFrame(
    {
        'a': [0, 0, 0, 0, 0, 954, 1],
        'b': [1000, 11123, 1123, 0, 55, 0, 1],
    },
)
python

3 个回答

  1. Best Answer

    您可以根据索引向掩码添加条件：

    mask = (df.a > df.b) & (df.index < 3)

    对于第二个示例，这将导致：

         a      b   c
0    0   1000 NaN
1    0  11123 NaN
2    0   1123 NaN
3    0      0 NaN
4    0     55 NaN
5  954      0 NaN
6    1      1 NaN
    • 2

  2. 如果您希望列的结果c与您的问题完全相同，那么您可以使用以下代码：

    mask = ((df.a > df.b) & (df.index < 3))
df["c"] = df.a[mask]

    当df.a[mask]在最后一行广播时，它将在掩码计算结果为 False 的所有行中放入 NaN。所以，这会导致：

          a      b       c
0   100   1000     NaN
1  1123  11123     NaN
2  9999   1123  9999.0
3   100      0     NaN
4     1     55     NaN
5   954      0     NaN
6     1      1     NaN
    • 2

  3. 无需计算另一个掩码，只需df['a']在分配期间进行切片和重新索引：

    mask = df['a'] > df['b']

df.loc[mask, 'c'] = df['a'].iloc[:3]

    输出：

    # example 1
      a      b       c
0   100   1000     NaN
1  1123  11123     NaN
2  9999   1123  9999.0
3   100      0     NaN
4     1     55     NaN
5   954      0     NaN
6     1      1     NaN

# example 2
     a      b   c
0    0   1000 NaN
1    0  11123 NaN
2    0   1123 NaN
3    0      0 NaN
4    0     55 NaN
5  954      0 NaN
6    1      1 NaN
    • 1

相关问题