Skip to main content

Seaborn Swarmplot "hue" not coloring correctly / as expected

When I graph my data with Seaborn swarmplot, it orders the overlapping points "middle out". Meaning, the larger levels are in the middle and the smaller are on the edges (like 1,1,2,2,1,1 or 2,2,3,4,2,2). This messes up the hue coloration as confirmed by getting the index of the point that I am hovering over and labeling it (see below images for proof). What I am unsure about is, is it my labeling method that is incorrect or is it Seaborn's hue that is messing up? I've tried reordering the dataframe used for the plot and also setting the hue_order but nothing has worked correctly.

Here is a snippet of the data:

import pandas as pd
from io import StringIO
rejectionDF = StringIO('''
ID,Level,Days_After
472,3,3
678,2,3
491,3,10
621,3,10
314,4,11
575,3,11
654,3,11
356,3,12
403,3,12
301,2,12
557,2,12
405,3,13
694,3,13
770,3,13
361,2,13
452,2,13
484,2,13
750,2,13
371,3,14
458,3,14
474,3,14
483,3,14
705,3,14
418,2,14
481,2,14
583,2,14
729,2,14
797,2,14
818,2,14
254,3,15
392,3,15
475,3,15
684,3,15
737,3,15
805,3,15
370,2,15
444,2,15
498,2,15
521,2,15
542,2,15
577,2,15
603,2,15
733,2,15
739,2,15
809,2,15
680,4,16
368,3,16
387,3,16
513,3,16
659,3,16
''')

rejectionDF = pd.read_csv(rejectionDF)

Here is the following code that I use:

%matplotlib notebook             # To show the hover in a Jupyter Notebook
import matplotlib.pyplot as plt
import seaborn as sns
import mplcursors


years = int(3)
timeframe = years*365 # 3 year time frame

# Unnecessary for the example, but I thought maybe sorting the data would work
rejection_timerange = rejectionDF[rejectionDF.Days_After <= timeframe].sort_values(by = ['Days_After', 'Level'], ascending = [True, False], ignore_index = True)

plt.rcParams["figure.figsize"] = (10,5)
plt.rcParams.update({'font.size': 10})
rej_fig, rej_ax = plt.subplots()
sns.set_palette(sns.color_palette(["orange", "orangered", "darkred"]))
rej_ax = sns.swarmplot(x = rejection_timerange.Days_After, y = [0]*len(rejection_timerange),
                       orient = "h", size=10, hue = rejection_timerange.Level, picker = 1)
@mplcursors.cursor(rej_ax, hover=2).connect("add")
def _(sel):
    ID = rejection_timerange.ID[sel.index]
    rejection_level = rejection_timerange.Level[sel.index]                                    
    color = rej_ax.collections[0].get_facecolors()[sel.index]
    sel.annotation.set_text(('Study ID: {}\nDays after: {}\nRejection Level: {}').format(ID, int(sel.target[0]), rejection_level))
    sel.annotation.get_bbox_patch().set(fc=color, alpha = 1)
    sel.annotation.arrow_patch.set(arrowstyle="-|>", connectionstyle="angle3", fc="black", alpha=.5)

Here are some incorrectly labeled or colored points:

enter image description here

enter image description here



source https://stackoverflow.com/questions/74197853/seaborn-swarmplot-hue-not-coloring-correctly-as-expected

Labels

Labels:

Comments

Post a Comment

Popular posts from this blog

ValueError: X has 10 features, but LinearRegression is expecting 1 features as input

So, I am trying to predict the model but its throwing error like it has 10 features but it expacts only 1. So I am confused can anyone help me with it? more importantly its not working for me when my friend runs it. It works perfectly fine dose anyone know the reason about it? cv = KFold(n_splits = 10) all_loss = [] for i in range(9): # 1st for loop over polynomial orders poly_order = i X_train = make_polynomial(x, poly_order) loss_at_order = [] # initiate a set to collect loss for CV for train_index, test_index in cv.split(X_train): print('TRAIN:', train_index, 'TEST:', test_index) X_train_cv, X_test_cv = X_train[train_index], X_test[test_index] t_train_cv, t_test_cv = t[train_index], t[test_index] reg.fit(X_train_cv, t_train_cv) loss_at_order.append(np.mean((t_test_cv - reg.predict(X_test_cv))**2)) # collect loss at fold all_loss.append(np.mean(loss_at_order)) # collect loss at order plt.plot(np.log(al...
Post a Comment
Read more

Sorting large arrays of big numeric stings

I was solving bigSorting() problem from hackerrank: Consider an array of numeric strings where each string is a positive number with anywhere from to digits. Sort the array's elements in non-decreasing, or ascending order of their integer values and return the sorted array. I know it works as follows: def bigSorting(unsorted): return sorted(unsorted, key=int) But I didnt guess this approach earlier. Initially I tried below: def bigSorting(unsorted): int_unsorted = [int(i) for i in unsorted] int_sorted = sorted(int_unsorted) return [str(i) for i in int_sorted] However, for some of the test cases, it was showing time limit exceeded. Why is it so? PS: I dont know exactly what those test cases were as hacker rank does not reveal all test cases. source https://stackoverflow.com/questions/73007397/sorting-large-arrays-of-big-numeric-stings
Post a Comment
Read more

How to load Javascript with imported modules?

I am trying to import modules from tensorflowjs, and below is my code. test.html <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Document</title </head> <body> <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.0.0/dist/tf.min.js"></script> <script type="module" src="./test.js"></script> </body> </html> test.js import * as tf from "./node_modules/@tensorflow/tfjs"; import {loadGraphModel} from "./node_modules/@tensorflow/tfjs-converter"; const MODEL_URL = './model.json'; const model = await loadGraphModel(MODEL_URL); const cat = document.getElementById('cat'); model.execute(tf.browser.fromPixels(cat)); Besides, I run the server using python -m http.server in my command prompt(Windows 10), and this is the error prompt in the console log of my browser: Failed to loa...
Post a Comment
Read more