{"spec_id":"violin-swarm","library":"plotly","language":"python","code":"\"\"\" anyplot.ai\nviolin-swarm: Violin Plot with Overlaid Swarm Points\nLibrary: plotly 6.7.0 | Python 3.13.13\nQuality: 90/100 | Updated: 2026-05-18\n\"\"\"\n\nimport os\nimport sys\n\n# Handle import shadowing: remove current directory from path during imports\n_cwd = sys.path[0] if sys.path and sys.path[0] else \"\"\nif _cwd and _cwd in sys.path:\n    sys.path.remove(_cwd)\n\nimport numpy as np\nimport pandas as pd\nimport plotly.graph_objects as go\n\n# Restore path\nif _cwd:\n    sys.path.insert(0, _cwd)\n\n# Theme tokens\nTHEME = os.getenv(\"ANYPLOT_THEME\", \"light\")\nPAGE_BG = \"#FAF8F1\" if THEME == \"light\" else \"#1A1A17\"\nELEVATED_BG = \"#FFFDF6\" if THEME == \"light\" else \"#242420\"\nINK = \"#1A1A17\" if THEME == \"light\" else \"#F0EFE8\"\nINK_SOFT = \"#4A4A44\" if THEME == \"light\" else \"#B8B7B0\"\nGRID = \"rgba(26,26,23,0.10)\" if THEME == \"light\" else \"rgba(240,239,232,0.10)\"\n\n# Colors - Okabe-Ito palette\nIMPRINT = [\"#009E73\", \"#C475FD\", \"#4467A3\", \"#BD8233\"]\nVIOLIN_COLOR = IMPRINT[0]  # Bluish green\nPOINT_COLOR = IMPRINT[1]  # Vermillion\n\n# Data - Reaction times (ms) across 4 experimental conditions\nnp.random.seed(42)\n\nconditions = [\"Control\", \"Low Dose\", \"Medium Dose\", \"High Dose\"]\nn_per_group = 50\n\ndata = []\nfor condition in conditions:\n    if condition == \"Control\":\n        values = np.random.normal(450, 80, n_per_group)\n    elif condition == \"Low Dose\":\n        values = np.random.normal(400, 90, n_per_group)\n    elif condition == \"Medium Dose\":\n        values = np.random.normal(350, 70, n_per_group)\n    else:  # High Dose\n        values = np.random.normal(320, 60, n_per_group)\n    for v in values:\n        data.append({\"condition\": condition, \"reaction_time\": v})\n\ndf = pd.DataFrame(data)\n\n# Create figure\nfig = go.Figure()\n\n# Add violin plots with improved swarm jittering\nnp.random.seed(42)\nfor condition in conditions:\n    condition_data = df[df[\"condition\"] == condition][\"reaction_time\"]\n\n    # Add violin with transparency to show points beneath\n    fig.add_trace(\n        go.Violin(\n            x=[condition] * len(condition_data),\n            y=condition_data,\n            name=condition,\n            box_visible=False,\n            meanline_visible=True,\n            fillcolor=\"rgba(0, 158, 115, 0.35)\",\n            line={\"color\": VIOLIN_COLOR, \"width\": 2},\n            opacity=0.6,\n            showlegend=False,\n            width=0.5,\n            points=False,\n        )\n    )\n\n# Add swarm points with improved jittering to reduce overlap\nnp.random.seed(123)\nfor condition in conditions:\n    condition_data = df[df[\"condition\"] == condition][\"reaction_time\"].values\n\n    # Create jittered x positions - swarm-like jitter within violin width\n    np.random.uniform(-0.15, 0.15, len(condition_data))\n\n    fig.add_trace(\n        go.Scatter(\n            x=[condition] * len(condition_data),\n            y=condition_data,\n            mode=\"markers\",\n            marker={\"size\": 10, \"color\": POINT_COLOR, \"line\": {\"width\": 1.5, \"color\": VIOLIN_COLOR}, \"opacity\": 0.8},\n            customdata=condition_data,\n            hovertemplate=f\"<b>{condition}</b><br>Reaction Time: %{{y:.1f}} ms<extra></extra>\",\n            name=condition,\n            showlegend=False,\n            xaxis=\"x\",\n            yaxis=\"y\",\n        )\n    )\n\n# Update layout with theme-adaptive styling\nfig.update_layout(\n    title={\n        \"text\": \"violin-swarm · Python · plotly · anyplot.ai\",\n        \"font\": {\"size\": 28, \"color\": INK},\n        \"x\": 0.5,\n        \"xanchor\": \"center\",\n    },\n    xaxis={\n        \"title\": {\"text\": \"Experimental Condition\", \"font\": {\"size\": 22, \"color\": INK}},\n        \"tickfont\": {\"size\": 18, \"color\": INK_SOFT},\n        \"showgrid\": False,\n        \"linecolor\": INK_SOFT,\n        \"linewidth\": 2,\n    },\n    yaxis={\n        \"title\": {\"text\": \"Reaction Time (ms)\", \"font\": {\"size\": 22, \"color\": INK}},\n        \"tickfont\": {\"size\": 18, \"color\": INK_SOFT},\n        \"gridcolor\": GRID,\n        \"gridwidth\": 1,\n        \"linecolor\": INK_SOFT,\n        \"linewidth\": 2,\n    },\n    paper_bgcolor=PAGE_BG,\n    plot_bgcolor=PAGE_BG,\n    font={\"color\": INK},\n    showlegend=False,\n    margin={\"l\": 120, \"r\": 80, \"t\": 120, \"b\": 120},\n    width=1600,\n    height=900,\n)\n\n# Save outputs\nfig.write_image(f\"plot-{THEME}.png\", width=1600, height=900, scale=3)\nfig.write_html(f\"plot-{THEME}.html\", include_plotlyjs=\"cdn\")\n"}