Helper function to compare two DataFrames and find rows which are unique or shared.

pandas_dataframe_difference.py

def dataframe_difference(df1, df2, which=None):
    """Find rows which are different."""
    comparison_df = df1.merge(df2,
                              indicator=True,
                              how='outer')
    if which is None:
        diff_df = comparison_df[comparison_df['_merge'] != 'both']
    else:
        diff_df = comparison_df[comparison_df['_merge'] == which]
    diff_df.to_csv('data/diff.csv')
    return diff_df