基于Pandas实现DataFrame成对行匹配与结果标记

import pandas as pd

data = {
    'Obs': [1, 2, 3, 4, 5, 6],
    'Dataset': ['Source', 'Target', 'Source', 'Target', 'Source', 'Target'],
    'Col1': ['A', 'A', 'B', 'B', 'C', 'D'],
    'Col2': [10, 10, 20, 20, 30, 30],
    'Col3': ['X', 'X', 'Y', 'Y', 'Z', 'Z']
}
df = pd.DataFrame(data)

print("原始DataFrame:")
print(df)

# 初始化'Result'列
df['Result'] = ''

# 分离Source和Target行
source_rows = df[df['Dataset'] == 'Source'].copy()
target_rows = df[df['Dataset'] == 'Target'].copy()

# 为Source和Target行创建临时的“对ID”，用于对齐
# 假设Source行索引为0, 2, 4...，Target行索引为1, 3, 5...
# 那么 (原始索引 // 2) 可以将每对Source/Target映射到同一个ID
source_rows['pair_id'] = source_rows.index // 2
target_rows['pair_id'] = target_rows.index // 2

# 基于pair_id合并Source和Target行，以便进行横向比较
# 这里使用 suffixes 来区分合并后的列名
merged_pairs = pd.merge(source_rows, target_rows, on='pair_id', suffixes=('_Source', '_Target'))

print("\n合并后的成对数据（用于比较）:")
print(merged_pairs)

# 定义需要比较的列
comparison_cols = ['Col1', 'Col2', 'Col3']

# 检查所有比较列是否都匹配
# (merged_pairs[[f'{col}_Source' for col in comparison_cols]].values == ...).all(axis=1)
# 这一步将生成一个布尔Series，指示每对数据是否完全匹配
all_cols_match = (merged_pairs[[f'{col}_Source' for col in comparison_cols]].values ==
                  merged_pairs[[f'{col}_Target' for col in comparison_cols]].values).all(axis=1)

print("\n每对数据是否完全匹配（布尔序列）:")
print(all_cols_match)

# 创建一个结果Series，索引对应原始DataFrame中的Source行
# merged_pairs的索引与source_rows的pair_id一致，我们需要将其转换回原始的df索引
source_original_indices = source_rows.index.values # 获取原始Source行的索引
results_series = pd.Series(
    ['Pass' if match else 'Fail' for match in all_cols_match],
    index=source_original_indices
)

# 更新原始DataFrame的'Result'列
df.loc[results_series.index, 'Result'] = results_series.values

print("\n更新Result列后的DataFrame:")
print(df)

# 获取当前列的顺序
cols = df.columns.tolist()

# 找到'Result'和'Dataset'列的索引
result_col_index = cols.index('Result')
dataset_col_index = cols.index('Dataset')

# 将'Result'列从当前位置移除
result_col = cols.pop(result_col_index)

# 将'Result'列插入到'Dataset'列之后
cols.insert(dataset_col_index + 1, result_col)

# 应用新的列顺序
df = df[cols]

print("\n最终结果DataFrame:")
print(df)

import pandas as pd

def determine_matching_pairs(df: pd.DataFrame, comparison_cols: list) -> pd.DataFrame:
    """
    根据成对的Source/Target行，比较指定列是否匹配，并标记结果。

    Args:
        df (pd.DataFrame): 包含Source和Target行的原始DataFrame。
                           假定Source和Target行交替出现。
        comparison_cols (list): 用于比较的列名列表。

    Returns:
        pd.DataFrame: 包含'Result'列（Pass/Fail）的更新后的DataFrame。
    """
    # 复制DataFrame以避免修改原始数据
    df_processed = df.copy()

    # 初始化'Result'列
    df_processed['Result'] = ''

    # 分离Source和Target行
    source_rows = df_processed[df_processed['Dataset'] == 'Source'].copy()
    target_rows = df_processed[df_processed['Dataset'] == 'Target'].copy()

    # 为Source和Target行创建临时的“对ID”，用于对齐
    source_rows['pair_id'] = source_rows.index // 2
    target_rows['pair_id'] = target_rows.index // 2

    # 基于pair_id合并Source和Target行，以便进行横向比较
    merged_pairs = pd.merge(source_rows, target_rows, on='pair_id', suffixes=('_Source', '_Target'))

    # 检查所有比较列是否都匹配
    # (merged_pairs[[f'{col}_Source' for col in comparison_cols]].values == ...)
    # .all(axis=1) 确保所有比较列在同一行都匹配
    all_cols_match = (merged_pairs[[f'{col}_Source' for col in comparison_cols]].values ==
                      merged_pairs[[f'{col}_Target' for col in comparison_cols]].values).all(axis=1)

    # 创建一个结果Series，索引对应原始DataFrame中的Source行
    source_original_indices = source_rows.index.values
    results_series = pd.Series(
        ['Pass' if match