仅显示列之间的差值小于30的行值

import pandas as pd

def filter_rows(dataset):
    result = []
    for i in range(len(dataset)):
        row = dataset.iloc[i]
        diff = abs(row[1] - row[0])  # 假设我们比较第一列和第二列的差值
        if diff < 30:
            result.append(row)
    return pd.DataFrame(result, columns=dataset.columns)

# 使用示例数据集
dataset = pd.DataFrame({'Column1': [10, 20, 30, 40, 50],
                        'Column2': [15, 25, 35, 45, 55],
                        'Column3': [20, 30, 40, 50, 60]})

# 调用函数进行筛选
filtered_dataset = filter_rows(dataset)

# 打印结果
print(filtered_dataset)