Python: How to find the number of items in each point on scatterplot and produce list?

Question

Right now I have a dataset of 1206 participants who have each endorsed a certain number of traumatic experiences and a number of symptoms associated with the trauma.

This is part of my dataframe (full dataframe is 1206 rows long):

SubjectID	PTSD_Symptom_Sum	PTSD_Trauma_Sum
1223	3	5
1224	4	2
1225	2	6
1226	0	3

question from:https://stackoverflow.com/questions/65889895/python-how-to-find-the-number-of-items-in-each-point-on-scatterplot-and-produce

Answer 1

You could create a new dataset with the counts of each pair 'PTSD_SUM', 'PTSD_Symptom_SUM' with:

counts = PTSD.groupby(by=['PTSD_symptom_SUM', 'PTSD_SUM']).size().to_frame('size').reset_index()

and then use Seaborn like this:

import seaborn as sns
sns.scatterplot(data=counts, x="PTSD_symptom_SUM", y="PTSD_SUM", hue="size", size="size")

To obtain something like this: