Discussion :

[Aylafreya]

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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.

[tyrtamos]

Bonjour,

Je suggère une solution de ce genre:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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

[Aylafreya]

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,

[tyrtamos]

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/.

[Aylafreya]

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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.