{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "collapsed": false
      },
      "outputs": [],
      "source": [
        "%matplotlib inline"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {},
      "source": [
        "\n# Batted Ball Trajectory - Initial Speed\n\n\n\nThis computes a set of trajectories at fixed launch angle, varying the initial speed.\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "collapsed": false
      },
      "outputs": [],
      "source": [
        "import pandas as pd\nimport seaborn as sns\nfrom matplotlib import pyplot as plt\n\nfrom pybbda.analysis.trajectories import BattedBallTrajectory\n\ntrajectory_calc = BattedBallTrajectory()\n\n\ntrajectories = pd.concat(\n    [\n        trajectory_calc.get_trajectory(\n            initial_speed=a,\n            launch_angle=35,\n            launch_direction_angle=0,\n            initial_spin=2500,\n            spin_angle=-10,\n        ).assign(launch_speed=\"ls=\" + str(a))\n        for a in range(10, 116, 25)\n    ],\n    axis=0,\n)\n\n\nplt.clf()\nsns.lineplot(\n    data=trajectories,\n    x=\"y\",\n    y=\"z\",\n    hue=\"launch_speed\",\n    palette=sns.cubehelix_palette(rot=-0.4)[0:5],\n)\n_ = plt.title(\"launch angle = 35\")"
      ]
    }
  ],
  "metadata": {
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "codemirror_mode": {
        "name": "ipython",
        "version": 3
      },
      "file_extension": ".py",
      "mimetype": "text/x-python",
      "name": "python",
      "nbconvert_exporter": "python",
      "pygments_lexer": "ipython3",
      "version": "3.7.3"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}