Violin plot on groupby'd series

[mwaskom]

Violin plot on groupby'd series

[mwaskom]

Are you sure you're using Pandas 0.12? I actually just ran into this problem today while testing against the Pandas 0.13 release candidate, but I think what you're doing should work with 0.12. In any case I was going to fix it and cut a version 0.2.1 in the next couple of days to address this and any other warts that pop up.

Here's a workaround, I think this is what the fix will look like too, just with lines 3 and 4 happening inside violinplot:

d = pd.Series(np.random.randn(100))
g = pd.Series(np.random.choice(list("abc"), 100))
gb = pd.groupby(d, g)
plot_data = [data for name, data in gb.groups.items()]
sns.violinplot(plot_data)

[olgabot]

Hmm on second thought I'm using the github master, which is closer to 0.13rc than 0.12. At first blush, it looks like that fix should work!

[mwaskom]

I just pushed a fix to the v0.2 branch, give it a shot when you have a chance and let me know if it fixes your issue.

[olgabot]

Still no dice, I'm getting a KeyError here:

    261         groups = pd.groupby(vals, groupby).groups
--> 262         vals = [vals.reindex(groups[name]) for name in order]

It works if the vals line is:

[vals.reindex(groups[group]) for group in groups]

I did that plus some other miscellaneous xlabel and names fixes here: #53 but there might be some residual code from my previous attempt at violinplots.

[mwaskom]

OK, I see what was wrong. The way I was doing things, it didn't recognize the dictionary style groupby object.

Because I didn't want to pile even more logic on top of the crappy duplicated code that set up the data for boxplot and violinplot, I ended up doing the long-awaited (#16) task of abstracting that code out into its own function and adding tests for it.

This now works on your data (Thanks Pandas :) ). By the way, I also transformed the data into tidy format (which is what I'm used to) to verify that the right thing was happening, in case that's useful.

[mwaskom]

Closing this as I merged #54 (I merged in v0.2 for the bugfix released and cherry-picked the commits into master) but feel free to reopen if you test and it's still broken.