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

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

.. _sphx_glr_auto_examples_sampling_alternatives.py:


List of alternatives
====================

Script reading the list of alternatives and identifying subsets

:author: Michel Bierlaire
:date: Mon Oct  9 10:53:03 2023

.. GENERATED FROM PYTHON SOURCE LINES 11-15

.. code-block:: default


    import pandas as pd
    from biogeme.partition import Partition


.. GENERATED FROM PYTHON SOURCE LINES 16-19

.. code-block:: default

    alternatives = pd.read_csv('restaurants.dat')
    alternatives


.. GENERATED FROM PYTHON SOURCE LINES 20-22

.. code-block:: default

    ID_COLUMN = 'ID'


.. GENERATED FROM PYTHON SOURCE LINES 23-25

.. code-block:: default

    all_alternatives = set(list(alternatives[ID_COLUMN]))


.. GENERATED FROM PYTHON SOURCE LINES 26-27

Set of Asian restaurants

.. GENERATED FROM PYTHON SOURCE LINES 27-30

.. code-block:: default

    asian = set(alternatives[alternatives['Asian'] == 1][ID_COLUMN])
    print(f'Number of asian restaurants: {len(asian)}')


.. GENERATED FROM PYTHON SOURCE LINES 31-32

Set of restaurants located in downtown

.. GENERATED FROM PYTHON SOURCE LINES 32-34

.. code-block:: default

    downtown = set(alternatives[alternatives['downtown'] == 1][ID_COLUMN])


.. GENERATED FROM PYTHON SOURCE LINES 35-36

Set of Asian restaurants in downtown

.. GENERATED FROM PYTHON SOURCE LINES 36-38

.. code-block:: default

    asian_and_downtown = asian & downtown


.. GENERATED FROM PYTHON SOURCE LINES 39-40

Set of Asian restaurants, and of restaurants in downtown

.. GENERATED FROM PYTHON SOURCE LINES 40-42

.. code-block:: default

    asian_or_downtown = asian | downtown


.. GENERATED FROM PYTHON SOURCE LINES 43-44

Set of Asian restaurants not in downtown

.. GENERATED FROM PYTHON SOURCE LINES 44-46

.. code-block:: default

    only_asian = asian - asian_and_downtown


.. GENERATED FROM PYTHON SOURCE LINES 47-48

Set of non Asian restaurants in downtown

.. GENERATED FROM PYTHON SOURCE LINES 48-50

.. code-block:: default

    only_downtown = downtown - asian_and_downtown


.. GENERATED FROM PYTHON SOURCE LINES 51-52

Set of restaurants that are neither Asian nor in downtown

.. GENERATED FROM PYTHON SOURCE LINES 52-55

.. code-block:: default

    others = all_alternatives - asian_or_downtown



.. GENERATED FROM PYTHON SOURCE LINES 56-61

.. code-block:: default

    def complement(a_set: set[int]) -> set[int]:
        """Returns the complement of a set"""
        return all_alternatives - a_set



.. GENERATED FROM PYTHON SOURCE LINES 62-63

Partitions.

.. GENERATED FROM PYTHON SOURCE LINES 63-73

.. code-block:: default

    partition_asian = Partition([asian, complement(asian)], full_set=all_alternatives)
    partition_downtown = Partition(
        [downtown, complement(downtown)], full_set=all_alternatives
    )
    partition_uniform = Partition([all_alternatives], full_set=all_alternatives)
    partition_uniform_asian = Partition([asian], full_set=asian)
    partition_uniform_asian_or_downtown = Partition(
        [asian_or_downtown], full_set=asian_or_downtown
    )


.. GENERATED FROM PYTHON SOURCE LINES 74-81

.. code-block:: default

    partitions = {
        'uniform': partition_uniform,
        'asian': partition_asian,
        'downtown': partition_downtown,
        'uniform_asian_or_downtown': partition_uniform_asian_or_downtown,
        'uniform_asian': partition_uniform_asian,
    }


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

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


.. _sphx_glr_download_auto_examples_sampling_alternatives.py:

.. only:: html

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




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

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

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

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


.. only:: html

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

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