如何将我的代码转换为一行？

问题描述

我有以下代码，我想使用列表理解将其转换为一行？但是我一直没有成功转换它。

exp_days = ["16/04/2021","23/04/2021","27/04/2021"]

for i in range(len(df)):
    if df["Date"][i] in exp_days:
        list_of_days.append(1)
    else:
        list_of_days.append(0)

我的数据框：

Date
16/04/2021
19/04/2021
20/04/2021
21/04/2021
22/04/2021
23/04/2021
26/04/2021
27/04/2021

预期输出：

list_of_days = [1,1,1]

解决方法

list_of_days = [ 1 if df["Date"][i] in exp_days else 0 for i in range(len(df)) ]

,

通过 numpy 替代 -

exp_days = ["16/04/2021","23/04/2021","27/04/2021"]
import numpy as np
result = np.where(df['Date'].isin(exp_days),1,0)