.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/montecarlo/swissmetro_one.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_montecarlo_swissmetro_one.py>`
        to download the full example code.

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

.. _sphx_glr_auto_examples_montecarlo_swissmetro_one.py:


Data preparation for Swissmetro: one observation
================================================

Use only the first observation for simulation.

Michel Bierlaire, EPFL
Tue Apr 29 2025, 12:20:15

.. GENERATED FROM PYTHON SOURCE LINES 11-17

.. code-block:: Python


    import pandas as pd

    from biogeme.database import Database
    from biogeme.expressions import Variable


.. GENERATED FROM PYTHON SOURCE LINES 18-20

.. code-block:: Python

    df = pd.read_csv('swissmetro.dat', sep='\t')


.. GENERATED FROM PYTHON SOURCE LINES 21-22

Use only the first observation (index 0)

.. GENERATED FROM PYTHON SOURCE LINES 22-25

.. code-block:: Python

    df = df.drop(df[df.index != 0].index)
    database = Database('swissmetro', df)


.. GENERATED FROM PYTHON SOURCE LINES 26-39

.. code-block:: Python

    SM_CO = Variable('SM_CO')
    TRAIN_CO = Variable('TRAIN_CO')
    CAR_CO = Variable('CAR_CO')
    TRAIN_TT = Variable('TRAIN_TT')
    SM_TT = Variable('SM_TT')
    CAR_TT = Variable('CAR_TT')
    GA = Variable('GA')
    CAR_AV = Variable('CAR_AV')
    TRAIN_AV = Variable('TRAIN_AV')
    SM_AV = Variable('SM_AV')
    SP = Variable('SP')
    CHOICE = Variable('CHOICE')


.. GENERATED FROM PYTHON SOURCE LINES 40-41

Definition of new variables

.. GENERATED FROM PYTHON SOURCE LINES 41-51

.. code-block:: Python

    SM_COST = SM_CO * (GA == 0)
    TRAIN_COST = TRAIN_CO * (GA == 0)
    CAR_AV_SP = CAR_AV * (SP != 0)
    TRAIN_AV_SP = TRAIN_AV * (SP != 0)
    TRAIN_TT_SCALED = TRAIN_TT / 100.0
    TRAIN_COST_SCALED = TRAIN_COST / 100
    SM_TT_SCALED = SM_TT / 100.0
    SM_COST_SCALED = SM_COST / 100
    CAR_TT_SCALED = CAR_TT / 100
    CAR_CO_SCALED = CAR_CO / 100


.. _sphx_glr_download_auto_examples_montecarlo_swissmetro_one.py:

.. only:: html

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

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

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

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

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

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: swissmetro_one.zip <swissmetro_one.zip>`


.. only:: html

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

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