msrigs_drawfigs1_usefulness.py

# Multi-Stage Re-Identification (MSRI) Game Solver (GS) v1.1
# Component: Showing results in scatter plot (privacy, and usefulness)
# © Oct 2021-2021 Zhiyu Wan, HIPLAB, Vanderbilt University
# Compatible with python 3.8.5. Package dependencies: Pandas 1.1.3, Matplotlib 3.3.1, Seaborn 0.11.0
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

pruning = 1
folder_result = 'Results2058/Violin/m2/'
if pruning == 1:
    folder_result += 'pruning/'
column_names = ['privacy', 'utility']
scenario_name = ['No-protection', 'K-anonymity', 'Random masking',
                 'Opt-in game', r'Masking game ($\alpha$=0)', r'Masking game ($\alpha$=1)']
privacy = [0.229, 0.972, 0.995, 0.949, 0.994, 0.990]
utility = [1, 0.898, 0.887, 0.902, 0.927, 0.948]

scenarios = scenario_name

dataset2 = pd.DataFrame({'Privacy': privacy, 'Usefulness': utility, 'Scenario': scenarios})
colors = ["tab:red", "tab:orange", "tab:green", "tab:cyan", "tab:blue", "tab:purple"]
# ‘tab:blue’,‘tab:orange’,‘tab:green’,‘tab:red’,‘tab:purple’,‘tab:brown’,‘tab:pink’,‘tab:gray’,‘tab:olive’,‘tab:cyan’
customPalette = sns.set_palette(sns.color_palette(colors))
markers = {"No-protection": "X", "K-anonymity": "^", "Random masking": "v", "Opt-in game": "o",
           r"Masking game ($\alpha$=0)": "s", r"Masking game ($\alpha$=1)": "D"}
ax = sns.scatterplot(data=dataset2, x='Usefulness', y='Privacy', s=100, hue='Scenario', style='Scenario', markers=markers,
                     palette=customPalette)
ax.legend(loc='lower left')
fig = plt.gcf()
fig.set_size_inches(5.25, 5.25)
plt.show()
ax.figure.savefig(folder_result + 'result_figure_S1.png', bbox_inches='tight',
                pad_inches=0.028, transparent=True, dpi=600)