使用 Pandas 的 isin 方法处理日期时出现 False 的问题-Python教程-PHP中文网

使用 Pandas 的 isin 方法处理日期时出现 False 的问题

霞舞

发布： 2025-10-12 11:54:42

原创

421人浏览过

使用 pandas 的 isin 方法处理日期时出现 false 的问题

本文旨在解决在使用 Pandas 的 `isin` 方法处理包含日期时间类型数据的 DataFrame 时，遇到的条件判断始终为 `False` 的问题。通过分析问题原因，提供解决方案，并给出最佳实践建议，帮助读者避免类似错误，提升数据处理效率。核心在于理解 `Timestamp.date` 的返回值，以及如何在 `isin` 方法中正确使用日期对象。

问题分析

当使用 Pandas 处理包含日期时间数据的 DataFrame 时，经常需要根据日期进行筛选或条件判断。isin 方法是常用的工具，但如果使用不当，可能会导致条件判断始终为 False，无法得到预期结果。

问题的核心在于 df["Record Date"].dt.date 返回的不是一个简单的日期对象，而是一个 "bound method Timestamp.date of Timestamp"。这意味着它是一个绑定到特定 Timestamp 对象的 date 方法，而不是方法执行后的结果。

解决方案

要解决这个问题，需要确保 isin 方法接收的是一个包含日期对象的列表或元组，而不是 Timestamp 对象的 date 方法。

以下是两种常见的解决方案：

1. 在 Plate 类定义中使用 date()

如果 plate.date 存储的是 Timestamp 对象，确保在比较之前将其转换为 date 对象。修改 Plate 类的 __init__ 方法，确保 self.date 存储的是 datetime.date 对象。

from datetime import datetime
from dateutil.parser import parse

class Plate:
    def __init__(self, ..., date=None):
        # ... 其他初始化代码 ...

        if date is not None:
            if isinstance(date, str):
                self.date = [parse(date).date()]  # Convert to list of date objects
            elif isinstance(date, list) or isinstance(date, tuple):
                if all((isinstance(item, str) or isinstance(item, datetime)) for item in date):
                    self.date = [parse(item).date() for item in date] # Convert to list of date objects
                else:
                    raise TypeError("The data type of the elements in the date list/tuple must be datetime or strings.")
            elif isinstance(date, datetime):
                self.date = [date.date()] # Convert to list of date objects
            else:
                raise TypeError("The data type of parameter date must be datetime.date, string (containing date) or list/tuple (of dates/strings).")

登录后复制

修改后的代码确保 plate.date 始终是一个包含 datetime.date 对象的列表。

问问小宇宙

问问小宇宙是小宇宙团队出品的播客AI检索工具

查看详情

2. 在 isin 方法中使用 date()

另一种方法是在使用 isin 方法时，显式地将 plate.date 中的元素转换为 datetime.date 对象。

import pandas as pd

if hasattr(plate, "date"):
    # Ensure plate.date is a list of datetime.date objects
    dates_to_check = [d.date() if isinstance(d, pd.Timestamp) else d for d in plate.date]
    condition = df["Record Date"].dt.date.isin(dates_to_check)
else:
    condition = df["Well Name"] != None # True for available data

df.loc[condition, ["sample_type", "index", "initial_measurement"]] = list((df.loc[condition, "Well Name"].astype(str).apply(get_sample_info)))

# Change the data types of the new columns
df = df.astype({"sample_type": str, "index": pd.Int64Dtype(), "initial_measurement": bool})

登录后复制

这段代码首先检查 plate.date 中的元素是否为 pd.Timestamp 对象，如果是，则调用 .date() 方法将其转换为 datetime.date 对象。

示例代码

以下是一个完整的示例，演示如何使用 Pandas 的 isin 方法正确处理日期：

import pandas as pd
from datetime import datetime

# 创建示例 DataFrame
data = {'Record Date': [datetime(2023, 12, 1), datetime(2023, 12, 6), datetime(2023, 12, 10)],
        'Well Name': ['A1', 'B2', 'C3']}
df = pd.DataFrame(data)

# 创建 Plate 对象
class Plate:
    def __init__(self, date=None):
        if date is not None:
            self.date = [date]
        else:
            self.date = []

plate = Plate(date=datetime(2023, 12, 1))

# 使用 isin 方法进行筛选
dates_to_keep = [d.date() if isinstance(d, datetime) else d for d in plate.date]
relevant_df = df[df["Record Date"].dt.date.isin(dates_to_keep)]

print(relevant_df)

登录后复制

这段代码首先创建了一个包含日期时间数据的 DataFrame，然后创建了一个 Plate 对象，并将其 date 属性设置为一个日期。最后，使用 isin 方法筛选出 Record Date 在 plate.date 中的行。