带有 if 条件循环的 python 列表理解

Question

如果我想根据另一列值的条件更改一列的值，是否可以对 dataframe 使用列表理解。

我希望使工作的代码是这样的：

return ['lower_level' for x in usage_time_df['anomaly'] if [y < lower_outlier for y in usage_time_df['device_years']]

谢谢！

Answer 1

我不认为你想做的事情可以在列表理解中完成，如果可以，它肯定不会有效率。

假设 dataframe usage_time_df有两列anomaly和device_years ，如果我理解正确，当lower_level中的值未达到lower_outlier （我猜是浮点数）时，您希望将anomaly中的值设置为device_years 。 这样做的自然方法是：

usage_time_df.loc[usage_time_df['device_years'] < lower_outlier, 'anomaly'] = 'lower_level'