Skip to content

Instantly share code, notes, and snippets.

@Thiagobc23
Last active August 13, 2022 20:22
Show Gist options
  • Save Thiagobc23/0bc5c5f530e8f8fd98710fe1ccb407ce to your computer and use it in GitHub Desktop.
Save Thiagobc23/0bc5c5f530e8f8fd98710fe1ccb407ce to your computer and use it in GitHub Desktop.
dummy data with a numerical and a categorical field
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D
import seaborn as sns
colors = {'bg':'#3D3C42', 'data':'#FEFBF6', 'data2':'#A6D1E6'}
# data
url = 'https://gist.githubusercontent.com/Thiagobc23/0bc5c5f530e8f8fd98710fe1ccb407ce/raw/4e084668a83ab9d0a0ace1425742835a0563bcef/quality.csv'
df = pd.read_csv(url)
df_unit = df.groupby('Unit').mean().reset_index()
# sort prod units by avg score
df_unit = df_unit.sort_values('Quality Score', ascending=False)
order = df_unit['Unit'].values.tolist()
# Style and Subplots
#sns.set_style("whitegrid")
fig, (ax, ax_leg) = plt.subplots(2, figsize=(12,10), facecolor=colors['bg'], gridspec_kw={'height_ratios':[6, 1]})
ax.set_facecolor(colors['bg'])
ax_leg.set_facecolor(colors['bg'])
# Plot jitter chart
sns.stripplot(y="Unit", x="Quality Score", data=df, order=order, jitter=0.2, size=4, alpha=0.8, color=colors['data'], ax=ax)
# Draw lines for mean value
ax.scatter(df_unit['Quality Score'], df_unit['Unit'], marker='|', s=2500, color=colors['data2'], alpha=1, zorder=99)
ax.scatter(df_unit['Quality Score']+2, df_unit['Unit'], marker='|', s=2500, color=colors['data2'], alpha=1, zorder=99)
ax.scatter(df_unit['Quality Score']-2, df_unit['Unit'], marker='|', s=2500, color=colors['data2'], alpha=1, zorder=99)
# Scales and Ticks
ax.set_ylim(len(order)-0.5, -0.5)
ax.set_xlim(0,1001)
xticks = np.arange(0,1001,100)
ax.set_xticks(xticks, color=colors['data'])
ax.set_yticks(order, color=colors['data'])
ax.tick_params(axis='both', which='major', labelsize=12, colors=colors['data'])
# Labels and Title
ax.set_title('Quality score by production unit\n', loc='left', fontsize=20, color=colors['data'])
ax.set_ylabel('')
ax.set_xlabel('')
#ax.set_xlabel('Score', fontsize=12, color=colors['data'])
# custom Legend
legend_elements = [Line2D([0], [0], marker='o', color=colors['bg'], label='One test result', markerfacecolor=colors['data'], markersize=10),
Line2D([0], [0], marker='|', color=colors['data2'], label='Mean quality score for the unit', linestyle='None', markersize=25)]
legend = ax_leg.legend(handles=legend_elements, loc='upper center', ncol=5, frameon=False)
plt.setp(legend.get_texts(), color=colors['data'], fontsize=12)
# clean second axis
ax_leg.set_xticks([])
ax_leg.set_yticks([])
sns.despine(fig, left=True, top=True, right=True, bottom=True)
plt.tight_layout()
plt.savefig('jitter.png', facecolor=colors['bg'])
Unit Quality Score
Alpha 625.33
Alpha 382.06
Alpha 494.15
Alpha 520.27
Alpha 703.93
Alpha 546.25
Alpha 196.36
Alpha 370.14
Alpha 488.22
Alpha 629.26
Alpha 589.58
Alpha 460.09
Alpha 734.15
Alpha 537.33
Alpha 507.95
Alpha 656.69
Alpha 666.96
Alpha 860.14
Alpha 495.86
Alpha 545.02
Alpha 615.65
Alpha 348.26
Alpha 476.02
Alpha 447.27
Alpha 541.62
Alpha 414.8
Alpha 680.9
Alpha 367.07
Alpha 509.36
Alpha 271.31
Alpha 603.63
Alpha 554.35
Alpha 738.62
Alpha 539.18
Alpha 661.96
Alpha 594.18
Alpha 662.6
Alpha 600.61
Alpha 328.28
Alpha 374.08
Alpha 662.65
Alpha 646.23
Alpha 667.02
Alpha 711.86
Alpha 657.12
Alpha 434.76
Alpha 592.41
Alpha 278.48
Alpha 633.66
Alpha 492.34
Alpha 418.42
Alpha 644.11
Alpha 614.59
Alpha 489.66
Alpha 536.63
Alpha 637.93
Alpha 506.45
Alpha 683.4
Alpha 736.51
Alpha 597.01
Alpha 555.32
Alpha 545.56
Alpha 909.81
Alpha 597.14
Alpha 557.83
Alpha 559.64
Alpha 504.21
Alpha 653.19
Alpha 543.16
Alpha 688.45
Alpha 727.09
Alpha 472.42
Alpha 638.49
Alpha 555.78
Alpha 485.68
Alpha 670.38
Alpha 562.49
Alpha 616.59
Alpha 593.09
Alpha 712.71
Alpha 633.95
Alpha 501.1
Alpha 606.75
Alpha 502.68
Alpha 535.51
Alpha 445.28
Alpha 590.32
Alpha 341.63
Alpha 530.67
Alpha 666.12
Alpha 420.23
Alpha 761.94
Alpha 454.36
Alpha 347.9
Alpha 582.85
Alpha 588.49
Alpha 504.93
Alpha 470.83
Alpha 447.26
Alpha 535.29
Alpha 684.21
Alpha 694.58
Alpha 577.25
Alpha 671.11
Alpha 867.21
Alpha 509.71
Alpha 578.63
Alpha 573.68
Alpha 618.47
Alpha 510.04
Alpha 594.17
Alpha 391.51
Alpha 450.73
Alpha 564.16
Alpha 392.7
Alpha 695.27
Alpha 495.07
Alpha 598.85
Alpha 543.66
Alpha 347.27
Alpha 633.78
Alpha 542.37
Alpha 513.23
Alpha 525.12
Alpha 483.42
Alpha 483.34
Alpha 563.06
Alpha 625.26
Alpha 485.7
Alpha 495.75
Alpha 326.61
Alpha 328.11
Alpha 463.45
Alpha 525.43
Alpha 399
Alpha 608.71
Alpha 534.54
Alpha 536.97
Alpha 481.45
Alpha 662.5
Alpha 654.32
Alpha 534.42
Alpha 438.14
Alpha 432.5
Alpha 593.97
Alpha 338.14
Alpha 515.76
Alpha 484.29
Alpha 524.18
Alpha 467.18
Alpha 487.1
Alpha 538.02
Alpha 500.13
Alpha 654.86
Alpha 467.35
Alpha 347.01
Alpha 573.96
Alpha 428.66
Alpha 361.85
Alpha 535.07
Alpha 690.37
Alpha 500.03
Alpha 562.26
Alpha 565
Alpha 628.87
Alpha 305.14
Alpha 555.6
Alpha 380.34
Alpha 644.59
Alpha 559.54
Alpha 347.16
Alpha 428.3
Alpha 371.34
Alpha 292.13
Alpha 657.63
Alpha 516.63
Alpha 539.02
Alpha 486.88
Alpha 400.27
Alpha 470.93
Alpha 333.03
Alpha 488.09
Alpha 688.56
Alpha 534.25
Alpha 456.38
Alpha 372.75
Alpha 554.39
Alpha 291.83
Alpha 378.34
Alpha 606.07
Alpha 749.31
Alpha 549.63
Alpha 619.35
Alpha 293.15
Alpha 488.51
Beta 618.63
Beta 436.08
Beta 263.83
Beta 464.76
Beta 434.78
Beta 405.43
Beta 484.78
Beta 573.98
Beta 567.89
Beta 531.29
Beta 379.14
Beta 755.67
Beta 577.76
Beta 595.59
Beta 582.81
Beta 295.88
Beta 468.66
Beta 285.01
Beta 602.59
Beta 408.76
Beta 590.37
Beta 794.79
Beta 466.13
Beta 579.08
Beta 425.79
Beta 579.04
Beta 569.11
Beta 600.59
Beta 390.53
Beta 550.01
Beta 671.33
Beta 539.51
Beta 398.33
Beta 502.84
Beta 807.29
Beta 548.58
Beta 602.01
Beta 557.01
Beta 774.68
Beta 540.7
Beta 564.18
Beta 459.4
Beta 735.42
Beta 533.79
Beta 347.11
Beta 565.69
Beta 602.79
Beta 462.9
Beta 531.61
Beta 578.21
Beta 533.11
Beta 945.67
Beta 590.33
Beta 662.32
Beta 443.76
Beta 675.52
Beta 571.88
Beta 397.34
Beta 440.51
Beta 679
Beta 593.94
Beta 654.6
Beta 357.33
Beta 630.2
Beta 540.97
Beta 477.72
Beta 544.5
Beta 683.71
Beta 528
Beta 555.12
Beta 429.44
Beta 484.4
Beta 548.98
Beta 517.74
Beta 323.62
Beta 542.31
Beta 647.31
Beta 549.09
Beta 638.79
Beta 388.84
Beta 528.68
Beta 675.31
Beta 616.8
Beta 580.67
Beta 416.93
Beta 391.37
Beta 411.61
Beta 509.34
Beta 412.25
Beta 795.21
Beta 286.79
Beta 397.46
Beta 511.22
Beta 632.84
Beta 379.71
Beta 544.42
Beta 553.93
Beta 428.09
Beta 373.77
Beta 243.16
Beta 697.77
Beta 549.77
Beta 528.04
Beta 536.26
Beta 470.54
Beta 643.84
Beta 311.28
Beta 385.56
Beta 544.48
Beta 475.77
Beta 643.94
Beta 497.83
Beta 422.72
Beta 221.69
Beta 314.03
Beta 597.21
Beta 543.35
Beta 580.07
Beta 394.81
Beta 623.99
Beta 522.99
Beta 742.11
Beta 524.2
Beta 674.13
Beta 335.82
Beta 387.13
Beta 614.99
Beta 471.86
Beta 590.95
Beta 417.81
Beta 580.38
Beta 791.27
Beta 413.88
Beta 532.35
Beta 474.52
Beta 313.6
Beta 685.49
Beta 608.15
Beta 486.72
Beta 527.89
Beta 452
Beta 369.6
Beta 500.95
Beta 243.25
Beta 608.27
Beta 486.5
Beta 609.55
Beta 397.74
Beta 361.02
Beta 463.02
Beta 487.62
Beta 628.32
Beta 635.37
Beta 356.51
Beta 434.99
Beta 548.23
Beta 492.31
Beta 486.92
Beta 434.48
Beta 560.83
Beta 615.78
Beta 586.58
Beta 385.15
Beta 531.92
Beta 484.55
Beta 440.62
Beta 597.68
Beta 272.52
Beta 480.53
Beta 570.55
Beta 417.29
Beta 424.7
Beta 492.66
Beta 334.66
Beta 572.51
Beta 418.21
Beta 477.63
Beta 413.8
Beta 496.21
Beta 465.55
Beta 667.52
Beta 551.41
Beta 331.8
Beta 462.11
Beta 525.43
Beta 467.75
Beta 508.32
Beta 651.36
Beta 438.9
Beta 284.52
Beta 408.97
Beta 542.63
Beta 569.15
Beta 472.8
Beta 531.15
Beta 605.59
Beta 413.21
Beta 569.47
Beta 404.39
Charlie 701.14
Charlie 623.38
Charlie 411.69
Charlie 476.36
Charlie 445.73
Charlie 596.96
Charlie 342.98
Charlie 405.83
Charlie 395.04
Charlie 578.31
Charlie 382.07
Charlie 695.27
Charlie 407.18
Charlie 564.37
Charlie 597.53
Charlie 497.02
Charlie 451.02
Charlie 581.46
Charlie 486.42
Charlie 634.16
Charlie 464.01
Charlie 426.75
Charlie 554.35
Charlie 547.55
Charlie 460.66
Charlie 434.14
Charlie 629.64
Charlie 358.27
Charlie 329.37
Charlie 354.5
Charlie 541.88
Charlie 319.94
Charlie 482.95
Charlie 489.27
Charlie 493.92
Charlie 554.31
Charlie 584.22
Charlie 397.84
Charlie 409.83
Charlie 486.35
Charlie 591.08
Charlie 335.6
Charlie 380.08
Charlie 735.24
Charlie 706.45
Charlie 520.59
Charlie 705.54
Charlie 434.86
Charlie 330.47
Charlie 430.4
Charlie 236.74
Charlie 327.59
Charlie 355.84
Charlie 431.94
Charlie 380.76
Charlie 623.06
Charlie 464.08
Charlie 363.73
Charlie 514.26
Charlie 744.87
Charlie 534.78
Charlie 253.66
Charlie 332.73
Charlie 410.23
Charlie 442.92
Charlie 687.43
Charlie 636.33
Charlie 582.44
Charlie 387.41
Charlie 346.35
Charlie 623.46
Charlie 470.56
Charlie 268.64
Charlie 497.53
Charlie 396.98
Charlie 494.91
Charlie 491.43
Charlie 390.45
Charlie 432.82
Charlie 474.39
Charlie 378.75
Charlie 579.69
Charlie 221.04
Charlie 474.95
Charlie 578.04
Charlie 467.36
Charlie 475.35
Charlie 406.76
Charlie 710.79
Charlie 690.02
Charlie 579.14
Charlie 555.6
Charlie 587.72
Charlie 551.65
Charlie 422.54
Charlie 493.09
Charlie 486.24
Charlie 203.09
Charlie 464.11
Charlie 442.81
Charlie 630.53
Charlie 293.17
Charlie 413.44
Charlie 460.42
Charlie 495.85
Charlie 779.36
Charlie 374.71
Charlie 456.97
Charlie 420.9
Charlie 337.63
Charlie 352.51
Charlie 576.38
Charlie 520.27
Charlie 497.75
Charlie 572.13
Charlie 333.95
Charlie 463.59
Charlie 572.95
Charlie 517.09
Charlie 320.73
Charlie 311.17
Charlie 606.82
Charlie 437.21
Charlie 674.35
Charlie 496.2
Charlie 425.19
Charlie 495.05
Charlie 286.31
Charlie 651.96
Charlie 518.6
Charlie 420.94
Charlie 480.31
Charlie 484.67
Charlie 645.9
Charlie 415.58
Charlie 593.37
Charlie 451.69
Charlie 400.64
Charlie 417.73
Charlie 266.94
Charlie 471.98
Charlie 569.81
Charlie 570.62
Charlie 610.78
Charlie 397.38
Charlie 769.78
Charlie 347
Charlie 493.87
Charlie 607.27
Charlie 448
Charlie 588.64
Charlie 618.26
Charlie 529.12
Charlie 339.38
Charlie 550.57
Charlie 653.72
Charlie 286.66
Charlie 637.82
Charlie 213.44
Charlie 564.11
Charlie 513.18
Charlie 325.39
Charlie 486.46
Charlie 552.55
Charlie 476
Charlie 561.51
Charlie 468.24
Charlie 372.02
Charlie 632.44
Charlie 434.37
Charlie 159.41
Charlie 336.43
Charlie 512.84
Charlie 462.85
Charlie 506.83
Charlie 479.55
Charlie 530.84
Charlie 687.6
Charlie 461.28
Charlie 534.86
Charlie 632.31
Charlie 260.62
Charlie 446.94
Charlie 338.03
Charlie 650.98
Charlie 512.16
Charlie 422.53
Charlie 452.5
Charlie 613.89
Charlie 556.21
Charlie 667.4
Charlie 438.28
Charlie 543.86
Charlie 399.42
Charlie 550.16
Charlie 621.26
Charlie 479.2
Charlie 637.63
Charlie 433.12
Charlie 472.61
Charlie 552.69
Charlie 489.47
Charlie 370.89
Charlie 360.02
Delta 318.33
Delta 444.47
Delta 327.86
Delta 424.39
Delta 467.33
Delta 252.62
Delta 448.54
Delta 445.9
Delta 392.5
Delta 547.95
Delta 437.91
Delta 422.89
Delta 451.06
Delta 801.91
Delta 411.37
Delta 534.44
Delta 421.96
Delta 488.07
Delta 494.34
Delta 364.09
Delta 548.57
Delta 468.76
Delta 359.22
Delta 378.17
Delta 512.32
Delta 402.11
Delta 178.25
Delta 424.61
Delta 483.3
Delta 427.5
Delta 375.81
Delta 435.28
Delta 436.5
Delta 361.86
Delta 350.41
Delta 353.92
Delta 315.84
Delta 574.35
Delta 463.66
Delta 506.28
Delta 290.08
Delta 393.16
Delta 396.66
Delta 384.92
Delta 324.82
Delta 412.69
Delta 396.16
Delta 262.7
Delta 398.32
Delta 569.41
Delta 514.09
Delta 563.1
Delta 457.26
Delta 366.18
Delta 383.57
Delta 334.61
Delta 424.1
Delta 351.31
Delta 473.37
Delta 301.95
Delta 389.55
Delta 245.36
Delta 545.47
Delta 362.7
Delta 559
Delta 328.31
Delta 494.35
Delta 647.51
Delta 311.73
Delta 235.88
Delta 330.25
Delta 550.19
Delta 514.38
Delta 497.78
Delta 278.35
Delta 763.85
Delta 412.21
Delta 222.74
Delta 370.34
Delta 457.61
Delta 123.41
Delta 523.72
Delta 705.32
Delta 451.03
Delta 569.07
Delta 503.44
Delta 186.06
Delta 452.62
Delta 490.43
Delta 386.04
Delta 470.88
Delta 455.04
Delta 754.9
Delta 327.72
Delta 584.54
Delta 412.63
Delta 483.27
Delta 660.86
Delta 418.14
Delta 569.94
Delta 463.29
Delta 516.43
Delta 552.13
Delta 552.18
Delta 604.18
Delta 567.02
Delta 473.45
Delta 392.22
Delta 460.56
Delta 751.56
Delta 581.45
Delta 407.53
Delta 413.7
Delta 520.47
Delta 472.61
Delta 419.54
Delta 563.49
Delta 454.18
Delta 585.91
Delta 556.17
Delta 331.72
Delta 614.93
Delta 764.97
Delta 333.25
Delta 317.21
Delta 482.94
Delta 268.28
Delta 450.16
Delta 648.14
Delta 199.95
Delta 616.65
Delta 375.95
Delta 487.51
Delta 486.73
Delta 500.78
Delta 660.99
Delta 442.55
Delta 527.36
Delta 630.41
Delta 429.83
Delta 343.09
Delta 394.71
Delta 696.48
Delta 530.59
Delta 429.32
Delta 524.97
Delta 580.19
Delta 444.62
Delta 660.89
Delta 558.98
Delta 426.41
Delta 556.16
Delta 561.86
Delta 511.08
Delta 241.31
Delta 524.41
Delta 264.32
Delta 567.06
Delta 484.2
Delta 510.37
Delta 222.47
Delta 577
Delta 605.97
Delta 369.92
Delta 402.65
Delta 473.83
Delta 622.48
Delta 600.18
Delta 560.18
Delta 571.86
Delta 631.05
Delta 519
Delta 445.1
Delta 422.45
Delta 347.21
Delta 558.29
Delta 572.47
Delta 582.4
Delta 751.51
Delta 448.17
Delta 457.38
Delta 368.5
Delta 435.49
Delta 630.51
Delta 373.09
Delta 439.84
Delta 187.03
Delta 382.7
Delta 374.25
Delta 352.52
Delta 269.92
Delta 233.14
Delta 866.49
Delta 575.59
Delta 446.5
Delta 362.03
Delta 556.92
Delta 725.08
Echo 500.27
Echo 290.91
Echo 533.88
Echo 447.22
Echo 298.15
Echo 455.89
Echo 519.01
Echo 491.71
Echo 703.33
Echo 331.45
Echo 479.89
Echo 496.81
Echo 550.3
Echo 713.41
Echo 422.47
Echo 270.32
Echo 409.01
Echo 366.37
Echo 659.22
Echo 557.77
Echo 604.53
Echo 440.2
Echo 592.36
Echo 376.66
Echo 579.65
Echo 741.8
Echo 685.28
Echo 580.25
Echo 592.72
Echo 477.14
Echo 575.9
Echo 596.52
Echo 465.68
Echo 588.82
Echo 671.13
Echo 460.04
Echo 519.03
Echo 435.24
Echo 513.4
Echo 644.93
Echo 431.5
Echo 548.77
Echo 328.59
Echo 405.72
Echo 363.9
Echo 599.32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment