df.query avec un string en filtre ne fonctionne pas

Version imprimable

06/01/2019, 16h43
AUREL42160

df.query avec un string en filtre ne fonctionne pas
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:

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!

@++
06/01/2019, 20h35
wiztricks

Salut,

Citation:

Envoyé par AUREL42160

Avez-vous une idée?

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

- W

Bonjour,

Ok, voici un exemple :

Code:

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!

@++

06/01/2019, 23h01
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
06/01/2019, 23h38
AUREL42160

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