python - 有条件 dropna() Pandas

Question

我有一个相当简单的问题:我想根据条件从 DataFrame 中删除行。数据框看起来像这样:

Program        act
Original RO     A
Original RO    nan
Followup RO     B
Followup RO    nan
Integral RO    nan

我只想删除Original RO 和Integral RO 程序的空值。所以它应该看起来像这样:

Program        act
Original RO     A
Followup RO     B
Followup RO    nan

当我尝试对数据帧的一部分进行某些操作并保持其余部分不变时，我总是遇到问题。

我尝试过这个:

df.loc[df.Program.str.match('^(Original|Integral)')] = df.dropna()

但是不起作用。我究竟做错了什么？提前致谢!

Answer 1

您可以使用 isin() 检查条件和 isna()并对数据框进行子集化。

lst = ['Original RO', 'Integral RO']

df = df[~(df['Program'].isin(lst) & df['act'].isna())]

# Another one using drop.
# df = df.drop(df[(df['Program'].isin(lst) & df['act'].isna())].index)

print (df)

       Program  act
0  Original RO    A
2  Followup RO    B
3  Followup RO  NaN