使用 Pandas 的 isin 方法进行日期匹配时返回 False 的问题解决

from datetime import datetime
from dateutil.parser import parse
import pandas as pd
import numpy as np

class Plate:
    def __init__(self, ..., date=None):

        # ... (其他初始化代码)

        if date is not None:
            if isinstance(date, str):
                self.date = [parse(date).date()]  # 将字符串解析为 datetime 对象，然后提取 date 部分
            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() if isinstance(item, str) else item.date() for item in date]  # 提取 date 部分
                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()]  # 提取 date 部分
            else:
                raise TypeError("The data type of parameter date must be datetime.date, string (containing date) or list/tuple (of dates/strings).")
        else:
            self.date = [] # 初始化为空列表，方便后续判断

# 示例数据
data = {'Record Date': pd.to_datetime(['2023-12-01 17:16:00', '2023-12-06 10:00:00', '2023-12-08 12:00:00']),
        'Well Name': ['A1', 'B2', 'C3']}
df = pd.DataFrame(data)

# 创建 Plate 对象
plate = Plate(date=['2023-12-01', '2023-12-06'])  # 使用字符串初始化 date

# 使用 isin 进行日期匹配
if hasattr(plate, "date") and plate.date: # 确保 plate.date 存在且不为空
    condition = df["Record Date"].dt.date.isin(plate.date)
else:
    condition = df["Well Name"] != None

df.loc[condition, ["sample_type", "index", "initial_measurement"]] = [('type1', 1, True), ('type2', 2, False), ('type3', 3, True)][:len(df.loc[condition])] # 简化赋值
df = df.astype({"sample_type": str, "index": pd.Int64Dtype(), "initial_measurement": bool})

print(df)