主页 / coding / 问题 / 78314515
Accepted
AmirX
Asked: 2024-04-12 14:24:34 +0800 CST

对包含掩码的第一个实例的数据帧进行切片的最佳方法是什么？

  • 772

这是我的数据框：

import pandas as pd
import numpy as np
df = pd.DataFrame(
    {
        'a': [np.nan, np.nan, np.nan, 20, 12, 42, 33, 32, 31],
        'b': [np.nan, np.nan, np.nan, np.nan, 2333, np.nan, np.nan, 12323, np.nan]
    }
)

掩码是：

mask = (
    (df.a.notna()) &
    (df.b.notna())
)

预期输出：切片df到 的第一个实例mask。请注意，第一行mask已包含：

      a        b
0   NaN      NaN
1   NaN      NaN
2   NaN      NaN
3  20.0      NaN
4  12.0   2333.0

第一个实例mask是 row 4。因此，将其分割到该索引就是目标。

这些是我的尝试。第一个有效，但我不确定该方法是否正确：

# attempt 1
idx = df.loc[mask.cumsum().eq(1) & mask].index[0]
df = df.loc[:idx]
print(df)
# attempt 2
out = df[~mask.cummax()]
python

1 个回答

  1. Best Answer

    添加DataFrame.shift到您的第二个解决方案：

    out = df[~mask.shift(fill_value=False).cummax()]
print (out)
      a       b
0   NaN     NaN
1   NaN     NaN
2   NaN     NaN
3  20.0     NaN
4  12.0  2333.0
    • 3

相关问题