比较熊猫中的两个列与数据框中的更新日期列

Question

我需要将列的一定范围与另一列中的特定字符串进行比较，以用1,7、30天更新日期列，

Col A    ColB           colc
5          Internal     1-1-2022
7          external     1-1-2022
4          external     1-1-2022
3          external     1-1-2022

因此，如果df [cola]＆lt; 8和df [colb] =外部向DF [Colc]添加7天。如果df [cola]＆gt; = 8 and df [colb] =外部将30天添加到df [colc]，以及df [colb] =内部将120天添加到DF [COLC]

Answer 1

您可以使用 numpy.select.select 处理所有条件：

我将它们简化为：

• 如果COLB为“内部” - ＆GT;加上120天
• ，如果可乐≤8-＆gt；加7天
• - ＆gt;添加30天的

factor = np.select([df['ColB'].eq('Internal'), df['Col A'].le(8)], [120, 7], 30)
df['new'] = (pd.to_datetime(df['colc'], dayfirst=True)
               .add(pd.DateOffset(days=1)*a)
               .dt.strftime('%-d-%-m-%-Y')
             )

输出：

   Col A      ColB      colc       new
0      5  Internal  1-1-2022  3-1-2022
1      7  external  1-1-2022  6-1-2022
2      4  external  1-1-2022  1-1-2022
3      3  external  1-1-2022  3-1-2022

Answer 2

使用相同的np.Select与@mozway，我建议使用pd.timedelta。

df = pd.DataFrame(
    {
        'Col A': [5, 7, 4, 3], 
        'ColB': ['Internal', 'external', 'external', 'external'], 
        'colc': pd.to_datetime("1/1/2022", format="%m/%d/%Y")
    }
)

df['colc'] = [
    (a + pd.Timedelta(b, 'd')) for (a, b) in
    zip(df.colc, np.select([df['ColB'].eq('Internal'), df['Col A'].le(8)], [120, 7], 30))
]

print(df)

结果：

   Col A      ColB       colc
0      5  Internal 2022-05-01
1      7  external 2022-01-08
2      4  external 2022-01-08
3      3  external 2022-01-08