{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": { "collapsed": false }, "outputs": [], "source": [ "import pandas as pd\n", "import numpy as np\n", "import matplotlib.pyplot as plt\n", "\n", "%load_ext autoreload\n", "%autoreload 2\n", "\n", "%matplotlib inline\n", "\n", "%precision 2\n", "pd.set_option('display.precision', 3)\n", "\n", "import ndl, sim\n", "from zt import ztnbinom" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/plain": [ "array([ 3.26e+03, 2.65e+03, 1.83e+03, 1.05e+03, 6.02e+02,\n", " 3.06e+02, 1.56e+02, 7.80e+01, 4.00e+01, 8.00e+00,\n", " 1.30e+01, 2.00e+00, 2.00e+00, 1.00e+00])" ] }, "execution_count": 2, "metadata": {}, "output_type": "execute_result" } ], "source": [ "ns = [ztnbinom.rvs(3,.6) for i in xrange(10000)]\n", "data = np.zeros((max(ns)))\n", "for i in ns:\n", " data[i-1] += 1\n", "data" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/html": [ "
\n", " | Frequency | \n", "Cues | \n", "Number | \n", "
---|---|---|---|
1 | \n", "3265 | \n", "[1, exactly1] | \n", "1 | \n", "
2 | \n", "2648 | \n", "[1, 2, exactly2] | \n", "2 | \n", "
3 | \n", "1832 | \n", "[1, 2, 3, exactly3] | \n", "3 | \n", "
4 | \n", "1047 | \n", "[1, 2, 3, 4, exactly4] | \n", "4 | \n", "
5 | \n", "602 | \n", "[1, 2, 3, 4, 5, exactly5] | \n", "5 | \n", "
6 | \n", "306 | \n", "[1, 2, 3, 4, 5, 6, exactly6] | \n", "6 | \n", "
7 | \n", "156 | \n", "[1, 2, 3, 4, 5, 6, 7, exactly7] | \n", "7 | \n", "
8 | \n", "78 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, exactly8] | \n", "8 | \n", "
9 | \n", "40 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, exactly9] | \n", "9 | \n", "
10 | \n", "8 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, exactly10] | \n", "10 | \n", "
11 | \n", "13 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, exactly11] | \n", "11 | \n", "
12 | \n", "2 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, exactl... | \n", "12 | \n", "
13 | \n", "2 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, ex... | \n", "13 | \n", "
14 | \n", "1 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14... | \n", "14 | \n", "
14 rows × 3 columns
\n", "\n", " | Cues | \n", "Number | \n", "Outcomes | \n", "
---|---|---|---|
1 | \n", "[1, exactly1] | \n", "1 | \n", "notdu | \n", "
2 | \n", "[1, 2, exactly2] | \n", "2 | \n", "du | \n", "
3 | \n", "[1, 2, 3, exactly3] | \n", "3 | \n", "notdu | \n", "
4 | \n", "[1, 2, 3, 4, exactly4] | \n", "4 | \n", "notdu | \n", "
5 | \n", "[1, 2, 3, 4, 5, exactly5] | \n", "5 | \n", "notdu | \n", "
6 | \n", "[1, 2, 3, 4, 5, 6, exactly6] | \n", "6 | \n", "notdu | \n", "
7 | \n", "[1, 2, 3, 4, 5, 6, 7, exactly7] | \n", "7 | \n", "notdu | \n", "
8 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, exactly8] | \n", "8 | \n", "notdu | \n", "
9 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, exactly9] | \n", "9 | \n", "notdu | \n", "
10 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, exactly10] | \n", "10 | \n", "notdu | \n", "
11 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, exactly11] | \n", "11 | \n", "notdu | \n", "
12 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, exactl... | \n", "12 | \n", "notdu | \n", "
13 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, ex... | \n", "13 | \n", "notdu | \n", "
14 | \n", "[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14... | \n", "14 | \n", "notdu | \n", "
14 rows × 3 columns
\n", "