如何将np.where（）应用于对数据帧的搜索？

Question

我到了dataframe：

• 出发 df_depart visualisation
• 数据排序单调 data_sorted_monotone visualisation

我想使用注释中的行作为条件：

np.where(
    #   df_depart['NOM SCADA'] in data_sorted_monotone.columns,
    df_depart['NOM SCADA'] =='test',
    'smth',
    np.nan
)

我想在df_depart中创建一个新行，这一行将由数据中的计算值填充。 为此，我需要知道“Varaible\u scada”列的值何时作为列名包含在data\u sorted\u monotone中。如果不是，我不会计算并填写“Nan”

有人有干净快捷的方法吗

谢谢

Answer 1

如果我知道你在说什么，请告诉我，但请尝试以下方法：

df_depart['new_col'] = np.where(df_depart['NOM SCADA'].isin(data_sorted_monotone.columns), 'smth', np.nan)