Pandas - Python，根据日期列删除行

Question

我正在尝试根据一个日期列删除数据框的行； [Delivery Date]

我需要删除超过 6 个月但不等于“1970”年的行。

我创建了 2 个变量：

from datetime import date, timedelta
sixmonthago = date.today() - timedelta(188)

import time
nineteen_seventy = time.strptime('01-01-70', '%d-%m-%y')

但我不知道如何使用[Delivery Date] 列根据这两个变量删除行。

谁能提供正确的解决方案？

Answer 1

确保计算本身在“6 个月”之前是准确的。您可能不想在 188 天内进行硬编码。并非所有月份都是平等的。

from datetime import date
from dateutil.relativedelta import relativedelta

#http://stackoverflow.com/questions/546321/how-do-i-calculate-the-date-six-months-from-the-current-date-using-the-datetime
six_months = date.today() - relativedelta( months = +6 )

那么你可以应用以下逻辑。

import time
nineteen_seventy = time.strptime('01-01-70', '%d-%m-%y')

df = df[(df['Delivery Date'].dt.year == nineteen_seventy.tm_year) | (df['Delivery Date'] >= six_months)]

如果您真的想删除数据框的部分，可以执行以下操作：

df = df[(df['Delivery Date'].dt.year != nineteen_seventy.tm_year) | (df['Delivery Date'] < six_months)].drop(df.columns)

Answer 2

你可以过滤掉它们：

df[(df['Delivery Date'].dt.year == 1970) | (df['Delivery Date'] >= sixmonthago)]

这将返回年份为 1970 年或日期小于 6 个月的所有行。

您可以使用布尔索引并传递多个条件来过滤 df，对于多个条件，您需要使用数组运算符所以| 而不是or，并且由于运算符优先级而在条件周围加上括号。

查看文档以获取boolean indexing 的解释