.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/indicators/optima_data.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_auto_examples_indicators_optima_data.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_indicators_optima_data.py:


Data preparation for Optima
===========================

Prepare data for the Optima case study.

:author: Michel Bierlaire
:date: Wed Apr 12 20:52:37 2023

.. GENERATED FROM PYTHON SOURCE LINES 12-17

.. code-block:: default


    import pandas as pd
    import biogeme.database as db
    from biogeme.expressions import Variable


.. GENERATED FROM PYTHON SOURCE LINES 18-19

Read the data

.. GENERATED FROM PYTHON SOURCE LINES 19-22

.. code-block:: default

    df = pd.read_csv('optima.dat', sep='\t')
    database = db.Database('optima', df)


.. GENERATED FROM PYTHON SOURCE LINES 23-24

Variables from the data

.. GENERATED FROM PYTHON SOURCE LINES 24-34

.. code-block:: default

    Choice = Variable('Choice')
    TimePT = Variable('TimePT')
    TimeCar = Variable('TimeCar')
    MarginalCostPT = Variable('MarginalCostPT')
    CostCarCHF = Variable('CostCarCHF')
    distance_km = Variable('distance_km')
    Gender = Variable('Gender')
    OccupStat = Variable('OccupStat')
    Weight = Variable('Weight')


.. GENERATED FROM PYTHON SOURCE LINES 35-36

Exclude observations such that the chosen alternative is -1

.. GENERATED FROM PYTHON SOURCE LINES 36-38

.. code-block:: default

    database.remove(Choice == -1.0)


.. GENERATED FROM PYTHON SOURCE LINES 39-40

Normalize the weights

.. GENERATED FROM PYTHON SOURCE LINES 40-43

.. code-block:: default

    sum_weight = database.data['Weight'].sum()
    number_of_rows = database.data.shape[0]
    normalized_weight = Weight * number_of_rows / sum_weight


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 0.000 seconds)


.. _sphx_glr_download_auto_examples_indicators_optima_data.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example




    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: optima_data.py <optima_data.py>`

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: optima_data.ipynb <optima_data.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_