Code source de ensae_teaching_cs.td_1a.construction_classique

# -*- coding: utf-8 -*-
"""
Quelques constructions classiques pour éviter de recoder des variantes d'algorithmes.
classiques.


:githublink:`%|py|7`
"""

from functools import reduce


[docs]def recherche(li, c): """ Retourne l'index d'un élément ou -1 si non trouvé. :param li: liste :param c: élément à trouver :return: position .. exref:: :tag: Base :title: recherche avec index Lorsqu'on cherche un élément dans un tableau, on cherche plus souvent sa position que le fait que le tableau contient cet élément. .. runpython:: :showcode: def recherche (li, c) : for i,v in enumerate (li) : if v == c: return i return -1 li = [45, 32, 43, 56] print (recherche(li, 43)) # affiche 2 En python, il existe un fonction simple qui permet de faire ça : :: print(li.index(43)) # affiche 2 Lorsque l'élément n'y est pas, on retourne souvent la position ``-1`` qui ne peut être prise par aucun élément : :: if c in li: return li.index(c) else: return -1 Même si ce bout de code parcourt deux fois le tableau (une fois déterminer sa présence, une seconde fois pour sa position), ce code est souvent plus rapide que la première version et la probabilité d'y faire une erreur plus faible. :githublink:`%|py|54` """ if c in li: return li.index(c) else: return -1
[docs]def minindex(li): """ Retourne l'index du minimum et le minimum. :param li: liste :return: tuple (minimum,position) .. exref:: :tag: Base :title: minimum avec position La fonction `min <https://docs.python.org/3/library/functions.html#min>`_ retourne le minium d'un tableau mais pas sa position. Le premier réflexe est alors de recoder le parcours de la liste tout en conservant la position du minimum. .. runpython:: :showcode: li = [0, 434, 43, 6436, 5] m = 0 for i in range (0, len(li)): if li[m] < li[i]: m = i print(m) Mais il existe une astuce pour obtenir la position sans avoir à le reprogrammer. .. runpython:: :showcode: li = [0, 434, 43, 6436, 5] k = [(v,i) for i, v in enumerate(li)] m = min(k) print(m) La fonction ``min`` choisit l'élément minimum d'un tableau dont les éléments sont des couples (élément du premier tableau, sa position). Le minimum est choisi en comparant les éléments, et la position départegera les exaequo. :githublink:`%|py|102` """ return min((v, i) for i, v in enumerate(li))
[docs]def recherche_dichotomique(li, c): """ Effectue une recherche dichotomique. :param li: tableau :param c: élément à chercher :return: position .. exref:: :tag: Base :title: recherche dichotomique La `recherche dichotomique <http://fr.wikipedia.org/wiki/Dichotomie>`_ est plus rapide qu'une recherche classique mais elle suppose que celle-ci s'effectue dans un ensemble trié. L'idée est de couper en deux l'intervalle de recherche à chaque itération. Comme l'ensemble est trié, en comparant l'élément cherché à l'élément central, on peut éliminer une partie de l'ensemble : la moitié inférieure ou supérieure. .. runpython:: :showcode: def recherche_dichotomique(li, c) : a, b = 0, len (li)-1 while a <= b : m = (a + b)//2 if c == li[m]: return m elif c < li[m]: b = m-1 else : a = m+1 return -1 print(recherche_dichotomique([0, 2, 5, 7, 8], 7)) :githublink:`%|py|138` """ a, b = 0, len(li) - 1 while a <= b: m = (a + b) // 2 if c == li[m]: return m elif c < li[m]: b = m - 1 # partie supérieure éliminée else: a = m + 1 # partie inférieure éliminée return -1 # élément non trouvé
[docs]def text2mat(s, sep_row="\n", sep_col="\t"): """ Convertit une chaîne de caractères en une matrice ( = liste de listes), réciproque de la fonction :func:`mat2text <ensae_teaching_cs.td_1a.construction_classique.mat2text>`. :param s: texte à convertir :param sep_row: séparation de ligne :param sep_col: séparateur de colonnes :return: liste de liste .. exref:: :tag: Base :title: conversion d'une chaîne de caractère en matrice Les quelques lignes qui suivent permettent de décomposer une chaîne de caractères en matrice. Chaque ligne et chaque colonne sont séparées par des séparateurs différents. Ce procédé intervient souvent lorsqu'on récupère des informations depuis un fichier texte lui-même provenant d'un tableur. .. runpython:: :showcode: s = "case11;case12;case13|case21;case22;case23" # décomposition en matrice ligne = s.split ("|") # lignes mat = [ l.split (";") for l in ligne ] # colonnes print(mat) Comme cette opération est très fréquente lorsqu'on travaille avec les données, on ne l'implémente plus soi-même. On préfère utiliser un module comme `pandas <http://pandas.pydata.org/>`_ qui est plus robuste et considère plus de cas. Pour écrire, utilise la méthode `to_csv <http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.to_csv.html>`_, pour lire, la fonction `read_csv <http://pandas.pydata.org/pandas-docs/stable/generated/pandas.io.parsers.read_csv.html>`_. On peut également directement enregistrer au format Excel `read_excel <http://pandas.pydata.org/pandas-docs/stable/generated/pandas.io.excel.read_excel.html>`_ et écrire dans ce même format `to_excel <http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.to_excel.html>`_. :githublink:`%|py|189` """ ligne = s.split(sep_row) # lignes mat = [li.split(sep_col) for li in ligne] # colonnes return mat
[docs]def mat2text(mat, sep_row="\n", sep_col="\t"): """ Convertit une matrice en une chaîne de caractères, réciproque de la fonction :func:`text2mat <ensae_teaching_cs.td_1a.construction_classique.text2mat>`. :param mat: matrice à convertir (liste de listes) :param sep_row: séparation de ligne :param sep_col: séparateur de colonnes :return: liste de liste .. exref:: :tag: Base :title: conversion d'une matrice en chaîne de caractères .. runpython:: :showcode: mat = [['case11', 'case12', 'case13'], ['case21', 'case22', 'case23']] ligne = [ ";".join (l) for l in mat ] # colonnes s = "|".join (ligne) # lignes print(s) :githublink:`%|py|217` """ ligne = [";".join(lc) for lc in mat] # colonnes s = "|".join(ligne) # lignes return s
[docs]def somme(li): """ Calcule la somme des éléments d'un tableau. :param li: tableau :return: somme .. exref:: :tag: Base :title: calcul d'une somme Le calcul d'une somme fait toujours intervenir une boucle car le langage :epkg:`Python` ne peut faire des additions qu'avec deux nombres. Le schéma est toujours le même : initialisation et boucle. .. runpython:: :showcode: li = [0, 434, 43, 6456] s = 0 # initialisation for l in li : # boucle s += l # addition print(s) Ce code est équivalent à la fonction `sum <https://docs.python.org/3/library/functions.html#sum>`_. Dans ce cas où la somme intègre le résultat d'une fonction (au sens mathématique) et non les éléments d'une liste, il faudrait écrire : .. runpython:: :showcode: def fonction(x): return x li = [0, 434, 43, 6456] s = 0 for l in li: s += fonction (l) print(s) Et ces deux lignes pourraient être résumées en une seule grâce à l'une de ces instructions : .. runpython:: :showcode: def fonction(x): return x li = [0, 434, 43, 6456] s1 = sum([fonction(l) for l in li]) s2 = sum(fonction(l) for l in li) s3 = sum(map(fonction, li)) print(s1, s2, s3) L'avantage des deux dernières instructions est qu'elles évitent la création d'une liste intermédiaire, c'est un point à prendre en compte si la liste sur laquelle opère la somme est volumineuse. :githublink:`%|py|282` """ return sum(li)
[docs]def triindex(li): """ Trie une liste, retourne la liste triée et les positions initiales. :param li: tableau :return: liste triée .. exref:: :tag: Base :title: tri, garder les positions initiales Le tri est une opération fréquente. On n'a pas toujours le temps de programmer le tri le plus efficace comme un tri `quicksort <http://fr.wikipedia.org/wiki/Tri_rapide>`_ et un tri plus simple suffit la plupart du temps. Le tri suivant consiste à recherche le plus petit élément puis à échanger sa place avec le premier élément du tableau du tableau. On recommence la même procédure à partir de la seconde position, puis la troisième et ainsi de suite jusqu'à la fin du tableau. .. runpython:: :showcode: li = [5, 6, 4, 3, 8, 2] for i in range (0, len (li)) : # recherche du minimum entre i et len (li) exclu pos = i for j in range (i+1, len (li)) : if li [j] < li [pos] : pos = j # échange ech = li [pos] li [pos] = li [i] li [i] = ech print(li) La fonction `sorted <https://docs.python.org/3/library/functions.html#sorted>`_ trie également une liste mais selon un algorithme plus efficace que celui-ci (voir `Timsort <http://en.wikipedia.org/wiki/Timsort>`_). On est parfois amené à reprogrammer un tri parce qu'on veut conserver la position des éléments dans le tableau non trié. Cela arrive quand on souhaite trier un tableau et appliquer la même transformation à un second tableau. Il est toujours préférable de ne pas reprogrammer un tri (moins d'erreur). Il suffit d'applicer la même idée que pour la fonction :func:`minindex <ensae_teaching_cs.td_1a.construction_classique.minindex>`. .. runpython:: :showcode: tab = ["zero", "un", "deux"] # tableau à trier pos = sorted( (t,i) for i,t in enumerate(tab) ) # tableau de couples print (pos) # affiche [('deux', 2), ('un', 1), ('zero', 0)] Si cette écriture est trop succincte, on peut la décomposer en : .. runpython:: :showcode: tab = ["zero", "un", "deux"] tab_position = [(t,i) for i,t in enumerate(tab)] tab_position.sort() print(tab_position) :githublink:`%|py|348` """ return sorted((t, i) for i, t in enumerate(li))
[docs]def compte(li): """ Compte le nombre d'occurrences de chaque élément d'une liste. :param li: tableau :return: dictionnaire .. exref:: :tag: Base :title: comptage On souhaite ici compter le nombre d'occurrences de chaque élément d'un tableau. Par exemple, on pourrait connaître par ce moyen la popularité d'un mot dans un discours politique ou l'étendue du vocabulaire utilisé. L'exemple suivant compte les mots d'une liste de mots. .. runpython:: :showcode: li = ["un", "deux", "un", "trois"] d = { } for l in li: if l not in d: d[l] = 1 else: d[l] += 1 print(d) # affiche {'un': 2, 'trois': 1, 'deux': 1} La structure la plus appropriée ici est un dictionnaire puisqu'on cherche à associer une valeur à un élément d'une liste qui peut être de tout type. Si la liste contient des éléments de type modifiable comme une liste, il faudrait convertir ceux-ci en un type immuable comme une chaîne de caractères. L'exemple suivant illustre ce cas en comptant les occurrences des lignes d'une matrice. .. runpython:: :showcode: mat = [ [1,1,1], [2,2,2], [1,1,1]] d = {} for l in mat: k = str(l) # k = tuple (l) lorsque cela est possible if k not in d: d[k] = 1 else: d[k] += 1 print(d) # affiche {'[1, 1, 1]': 2, '[2, 2, 2]': 1} Les listes ne peuvent pas être les clés du dictionnaire : `Why Lists Can't Be Dictionary Keys <https://wiki.python.org/moin/DictionaryKeys>`_. On peut également vouloir non pas compter le nombre d'occurrence mais mémoriser les positions des éléments tous identiques. On doit utiliser un dictionnaire de listes : .. runpython:: :showcode: li = ["un", "deux", "un", "trois"] d = { } for i, v in enumerate(li): if v not in d: d[v] = [i] else: d[v].append(i) print(d) # affiche {'un': [0, 2], 'trois': [3], 'deux': [1]} S'il suffit juste de compter, l'écriture la plus simple est : .. runpython:: :showcode: r = {} li = ["un", "deux", "un", "trois"] for x in li: r[x] = r.get(x,0) + 1 print(r) :githublink:`%|py|427` """ r = {} for x in li: r[x] = r.get(x, 0) + 1 return r
[docs]def mat2vect(mat): """ Convertit une matrice en un tableau à une seule dimension, réciproque de la fonction :func:`vect2mat <ensae_teaching_cs.td_1a.construction_classique.vect2mat>`. :param mat: matrice :return: liste .. exref:: :tag: Base :title: conversion d'une matrice en un vecteur Dans un langage comme le *C++*, il arrive fréquemment qu'une matrice ne soit pas représentée par une liste de listes mais par une seule liste car cette représentation est plus efficace. Il faut donc convertir un indice en deux indices ligne et colonne. Il faut bien sûr que le nombre de colonnes sur chaque ligne soit constant. Le premier programme convertit une liste de listes en une seule liste. .. runpython:: :showcode: mat = [[0,1,2],[3,4,5]] lin = [ i * len (mat [i]) + j \\ for i in range (0, len (mat)) \\ for j in range (0, len (mat [i])) ] print(lin) Vous pouvez aussi utiliser des fonctions telles que `reduce <https://docs.python.org/3/library/functools.html?highlight=reduce#module-functools>`_. .. runpython:: :showcode: from functools import reduce mat = [[0,1,2], [3,4,5]] lin = reduce(lambda x,y: x+y, mat) print(lin) :githublink:`%|py|471` """ return reduce(lambda x, y: x + y, mat)
[docs]def vect2mat(vect, ncol): """ Convertit un tableau à une dimension en une matrice, réciproque de la fonction :func:`mat2vect <ensae_teaching_cs.td_1a.construction_classique.mat2vect>`. :param vect: vecteur :param ncol: nombre de colonnes :return: matrice .. exref:: :tag: Base :title: conversion d'un vecteur en une matrice Dans un langage comme le *C++*, il arrive fréquemment qu'une matrice ne soit pas représentée par une liste de listes mais par une seule liste car cette représentation est plus efficace. Il faut donc convertir un indice en deux indices ligne et colonne. Il faut bien sûr que le nombre de colonnes sur chaque ligne soit constant. Le premier programme convertit une liste de listes en une seule liste. .. runpython:: :showcode: ncol = 2 vect = [0, 1, 2, 3, 4, 5] mat = [vect[i*ncol: (i+1)*ncol] for i in range(0,len(vect)//ncol)] print(mat) :githublink:`%|py|502` """ return [vect[i * ncol: (i + 1) * ncol] for i in range(0, len(vect) // ncol)]
[docs]def integrale(fonction, a, b, n): """ Calcule l'intégrale d'une fonction avec la `méthode de Rienmann <https://fr.wikipedia.org/wiki/Somme_de_Riemann>`_. :param fonction: fonction :param a: borne inférieure de l'intervalle :param b: borne supérieure de l'intervalle :param n: nombre de division de l'intervalle :return: valeur .. exref:: :tag: Base :title: fonction comme paramètre Une fonction peut aussi recevoir en paramètre une autre fonction. L'exemple suivant inclut la fonction ``calcul_n_valeur`` qui prend comme paramètres ``l`` et ``f``. Cette fonction calcule pour toutes les valeurs ``x`` de la liste ``l`` la valeur ``f(x)``. ``fonction_carre`` ou ``fonction_cube`` sont passées en paramètres à la fonction ``calcul_n_valeur`` qui les exécute. .. runpython:: :showcode: def fonction_carre(x): return x*x def fonction_cube(x): return x*x*x def calcul_n_valeur(l,f): res = [f(i) for i in l] return res l = [0,1,2,3] print(l) # affiche [0, 1, 2, 3] l1 = calcul_n_valeur(l, fonction_carre) print(l1) # affiche [0, 1, 4, 9] l2 = calcul_n_valeur(l, fonction_cube) print(l2) # affiche [0, 1, 8, 27] :githublink:`%|py|551` """ h = (b - a) / n return sum(fonction(a + h / 2 + h * i) for i in range(0, n)) * h
[docs]def construit_matrice_carree(n): """ Cette fonction construit une matrice carrée remplie de zéro sous la forme d'une liste de listes. :param n: dimension de la matrice carrée :githublink:`%|py|562` """ return [[0 for i in range(n)] for j in range(n)]
[docs]def enumerate_permutations_recursive(ensemble): """ Enumère les permutations d'un ensemble de façon récursive. :param ensemble: ensemble à permuter :return: itérateur sur les permutations :githublink:`%|py|572` """ if len(ensemble) == 1: yield ensemble else: for i in range(0, len(ensemble)): ensemble[0], ensemble[i] = ensemble[i], ensemble[0] per = enumerate_permutations_recursive(ensemble[1:]) for p in per: yield [ensemble[0]] + p ensemble[0], ensemble[i] = ensemble[i], ensemble[0]
[docs]def enumerate_permutations(ensemble): """ Enumère les permutations d'un ensemble de façon non récursive. :param ensemble: ensemble à permuter :return: itérateur sur les permutations :githublink:`%|py|591` """ if len(ensemble) == 1: yield ensemble else: position = list(range(len(ensemble))) while position[0] < len(ensemble): memo = [] for i, p in enumerate(position): ensemble[i], ensemble[p] = ensemble[p], ensemble[i] memo.append((i, p)) yield ensemble.copy() for i, p in reversed(memo): ensemble[i], ensemble[p] = ensemble[p], ensemble[i] last = len(position) - 1 position[last] += 1 while last > 0 and position[last] >= len(position): position[last - 1] += 1 position[last] = last last -= 1