Discussion :

[AUREL42160]

Bonjour,

J'ai crée une fonction qui utilise df.query pour filtrer un dataframe.
Je créer le filtre que je souhaite en string avec des conditions :

Code :

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

1
2
3
4
5
6
 
def mafonction(arg1, arg2, arg3...)
maselection = ''
if arg1 = aa :
  maselection = maselection + xxx
...

J'arrive à avoir le résultat souhaité avec la suite "A == 'eee' & B == 'rrr'"

Dès que que met la valeur renvoyée de maselection dans df.query() en faisant un copier-coller cela fonctionne et j'obtiens le dataframe filtré.
Mais dès que je veux utiliser df.query(maselection), cela me renvoi vers une message d'erreur en lien avec le fichier necompiler.py : unknown type str544...

Avez-vous une idée?

L'idée de base est de filtrer un dataframe avec un filtre sur plusieurs colonnes en même temps sans que tous les arguments de la fonction soient obligatoirement tous appelés...
Si vous avez une meilleures solution à proposer, je suis preneur également!

@++

[wiztricks]

Salut,

Avez-vous une idée?

Il va falloir un peu plus de code pour permettre de reproduire le problème rencontré..

- W

[AUREL42160]

Bonjour,

Ok, voici un exemple :

Code :

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

1
2
3
4
5
6
7
8
9
10
11
ar = numpy.array([['C', 'E', 'A.I',' U'], ['H',' O', 'N', 'Y'], ['K', 'P', 'HPP', 'TYR']])
df = pandas.DataFrame(ar, index = ['a1', 'a2', 'a3'], columns = ['A', 'B', 'C', 'D'])
 
def mafonction(arg1, arg2):
    maselection = '"'
    if arg1 != '*':
        maselection = maselection + "A == '" + arg1 + "' & "
    if arg2 !=  '*':
        maselection = maselection + "B == '" + arg2 + "'"
    maselection = maselection + '"'
    return df.query(maselection)

mafonction('C','E') renvoie une erreur alors que df.query("A == 'C' & B == 'E'") renvoie bien un df trié...

En espérant avoir été plus clair!

@++

[wiztricks]

Mouais...
Dans votre fonction, vous vous appliquez à ajouter un '"' au début et un '"' à la fin.
Je ne vois pas trop pourquoi vous faites çà mais là est la différence qui cause problème: supprimez les et çà ira mieux.

- W

[AUREL42160]

En effet, merci!
Je ne savais pas qu'avec une variable on devait enlever les guillemets...