使用df.query()构造DataFrame过滤查询字符串的问题求助

问题场景 #

需要将两个列表拼接成可用于df.query()的查询条件字符串：

• 条件列表：

  list1 = ['Product == "Dress"', 'Product == "Shoes"', 'City == "Mumbai"']
  

• 逻辑运算符列表：

  list2 = ['|', '&']
  

期望得到的查询字符串：

new_string = 'Product == "Dress" | Product == "Shoes" & City == "Mumbai"'

现有代码问题 #

你的嵌套join写法会让每个条件都与所有运算符拼接，导致输出出现重复运算符和多余内容，完全不符合需求。

正确解法 #

由于条件数量比运算符多1（n个条件需要n-1个运算符），可以将前n-1个条件与运算符一一配对拼接，最后加上最后一个条件：

# 配对前两个条件和对应的运算符，每个条件后拼接运算符和空格
parts = [f"{cond} {op}" for cond, op in zip(list1[:-1], list2)]
# 拼接所有部分并加上最后一个条件
new_string = ''.join(parts) + list1[-1]
print(new_string)

运行后将得到符合预期的输出：

Product == "Dress" | Product == "Shoes" & City == "Mumbai"

内容的提问来源于stack exchange，提问作者Roshini Prashanthica