Renommage champs dans un tableau - Optimisation ?

Version imprimable

1 pièce(s) jointe(s)

Bonjour,

J'ai écrit un bout de code dont le but est de modifier les champs d'une colonne, en fonction de leurs valeurs (Par exemple, un "RED" et un "R" vont devenir "Red").
C'est très basique, par contre, je n'ai pas trouvé le moyen de l'optimiser en l'écrivant en utilisant moins de lignes (avec des "or" ou autres). Du coup, j'ai dupliqué bêtement mes lignes pour chaque valeur. Est-ce que vous auriez une idée d'optimisation de ce code ?

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# On homogéneise le nom de la colonne Colour_raw
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'RED'] = 'Red'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'R'] = 'Red'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'red'] = 'Red'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'BLUE'] = 'Blue'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Bleu'] = 'Blue'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Blue'] = 'Blue'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'B'] = 'Blue'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'G'] = 'Green'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'GREEN'] = 'Green'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Green'] = 'Green'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'green'] = 'Green'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'O'] = 'Orange'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Orang'] = 'Orange'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Orange'] = 'Orange'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'PURP'] = 'Purple'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Purpal'] = 'Purple'
df_config['Colour_raw'].loc[df_config['Colour_raw'] == 'Purple'] = 'Purple'

Merci par avance pour votre aide.

PS.:
A toutes fin utiles, je joints le code général dont il est issue.

Bonjour,

Je suggère une solution de ce genre:
Code:

1 2 3 4 5 6 7 8 9 dico = {"R":"Red", "B":"Blue", "G":"Green", "O":"Orange", "P":"Purple"} ch = "bl" try: ch = dico[ch.upper()[0]] except KeyError: print("Erreur: la clé n'existe pas") except IndexError: print("Erreur: la chaine est vide") print(ch)
Ce qui affiche ici: "Blue"

Bien sûr, si on est sûr que la chaine ne contient que les 1ères lettres et n'est jamais vide, on peut se passer du try..except, ce qui simplifie le code:
Code:

1 2 3 4 dico = {"R":"Red", "B":"Blue", "G":"Green", "O":"Orange", "P":"Purple"} ch = "bl" ch = dico[ch.upper()[0]] print(ch)

01/02/2019, 15h38
Aylafreya

Merci...mais

Bonjour,

Merci pour la réponse. En effet, on obtient bien la valeur souhaitée (ex. red) quelquesoit le format rentré (R, r, etc.).
Par contre, j'ai essayé de l'appliquer à l'ensemble de ma colonne "Colour_raw" et pour le moment je n'ai pas trouvé le moyen. Est-ce qu'il faut que je boucle sur chaque ligne pour faire la comparaison ou y a t'il un moyen de l'appliquer à toute la colonne sans boucler?
Pour le moment, je me retrouve avec ce cractère "ch" dont je ne sais pas trop quoi faire au niveau de mon tableau.

Désolé, mais je suis dans mes tous premiers pas ....

Merci par avance.

Cdlt,
01/02/2019, 20h40
tyrtamos

Citation:

Envoyé par Aylafreya

Pour le moment, je me retrouve avec ce cractère "ch" dont je ne sais pas trop quoi faire au niveau de mon tableau.

ch est ici le nom quelconque d'une variable qui pointe sur la chaine de caractère qu'on veut changer ("r" => "Red"). Vous la remplacez par la bonne variable de votre code.

Cependant, je n'utilise ni pandas ni numpy, et de ce fait, il me manque trop d'éléments sur leur outils pour vous aider plus.

Mais votre question sur "ch" me dit que vous avez absolument besoin d'en connaitre un peu plus sur Python. A défaut, vous allez buter trop souvent sur les bases du langage. Je vous recommande le cours ici: https://python.developpez.com/cours/apprendre-python3/.

Résolu

Bonjour,

OK, j'ai répondu un peu rapidement, oui j'ai compris que Ch est une variable.
Merci pour le lien vers le cours, il est intéressant.
Pour le coup, j'ai trouvé la solution par un autre biais, au final c'est très basique
Code:

1 2 3 4 df_config_final = df_config['Colour_raw'].replace({'R': 'Red','RED' : 'Red', 'red': 'Red', 'BLUE' : 'Blue', 'Bleu' : 'Blue', 'B': 'Blue','G' : 'Green', 'GREEN': 'Green', 'green': 'Green', 'O' : 'Orange', 'Orang' : 'Orange', 'PURP': 'Purple','Purpal' : 'Purple'})
Merci pour le temps consacré et bonne journée.