在2个数据框架之间切换行

Question

我有2个看起来像这样的数据范围：

df1 = pd.DataFrame({'col1': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
       'col2': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
       'col3': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
       'cond': [5000, 6000, 6001, 5000, 6002, 6003, 5000, 6004, 6005, 5001]})
df2 = pd.DataFrame({'col1': [10, 9, 8, 7, 6, 5, 4, 3, 2, 1],
       'col2': [10, 9, 8, 7, 6, 5, 4, 3, 2, 1],
       'col3': [10, 9, 8, 7, 6, 5, 4, 3, 2, 1],
       'cond': [6000, 5000, 5001, 6000, 5002, 5003, 6000, 5004, 5005, 6001]})

我需要根据条件列在两个数据范围之间交换行，5000±100的值应在一个数据范围内，而值6000±100在另一个数据范围内：

df_expected1 = pd.DataFrame({'col1': [1, 9, 8, 4, 6, 5, 7, 3, 2, 10],
                'col2': [1, 9, 8, 4, 6, 5, 7, 3, 2, 10],
                'col2': [1, 9, 8, 4, 6, 5, 7, 3, 2, 10],
                'cond': [5000, 5000, 5001, 5000, 5002, 5003, 5000, 5004, 5005, 5001]})

df_expected2 = pd.DataFrame({'col1': [10, 2, 3, 7, 5, 6, 4, 8, 9, 1],
                'col2': [10, 2, 3, 7, 5, 6, 4, 8, 9, 1],
                'col2': [10, 2, 3, 7, 5, 6, 4, 8, 9, 1],
                'cond': [6000, 6000, 6001, 6000, 6002, 6003, 6000, 6004, 6005, 6001]})

Answer 1

首先将两个数据范围串联，然后基于列cond：

df = pd.concat([df1, df2]).sort_index()

df_expected1 = df[df.cond < 5500]
df_expected2 = df[df.cond > 5500]

我在这里假设df1和df2是数据范围。值5500是任意选择的，也许您需要更改此内容。

如果输出需要是字典，则可以使用.to_dict（）。