处理NP。

Question

我有一个变量“ date_0”，该变量对应于特定周期内的感染日期，并且我必须从上一个变量（date_0）使用日期条件（＆gt; = 33天）创建另一个变量“ date_1”。 “ date_0”和“ date_1”的格式为％y-％m-％d％h：％m：％s。 因此，基本上我应该执行这样的日期条件：

如果DF ['date_1]＆gt; = 33天，则df ['date_1] else df ['date_0']

，我这样做：

df ['date_1'] = np.Where（（（df ['date_1']＆gt; = 33d），df ['date_1']，df ['date_o']）

，我也尝试：

<代码> df ['date_1'] = np.Where（（DF ['date_1']。days＆gt; = 33），df ['date_1']，df ['date_o']）

，我失败了请帮我吗？

Answer 1

您可以使用相当于np.where的pandas。前任：

import pandas as pd

df = pd.DataFrame({'date_0': [pd.Timestamp('2022-01-01'), pd.Timestamp('2022-02-01')],
                   'date_1': [pd.Timestamp('2022-01-29'), pd.Timestamp('2022-03-15')]})

df['days_diff'] = (df['date_1']-df['date_0']).dt.days

# date_select:
# use date 1 if the difference is greater equal 33 days else use date 0
df['date_select'] = df['date_1'].where(df['days_diff'] >= 33, df['date_0'])

df
      date_0     date_1  days_diff date_select
0 2022-01-01 2022-01-29         28  2022-01-01
1 2022-02-01 2022-03-15         42  2022-03-15