如何简化具有多个包含变量的 .str.contains 代码？

Question

我正在使用 Series.str.contains() 函数来查找字典中的值是否包含在特定列的字符串中。我的代码有效，但我正在尝试简化它。

我尝试使用 for 循环遍历列表中的值。我的尝试在下面。

 total_joined_list=[joined_groc_lst,joined_util_lst]

 for i in range(len(total_joined_list)):



groc_amount,util_amount=abs(round(df.loc[df['Description'].str.contains('\b'+i+'\b',na=False),'Amount'].sum(),2))

这是当前的工作代码。

 joined_groc_lst = '|'.join(expenses_dict['Groceries'])
 joined_util_lst = '|'.join(expenses_dict['Utilities'])



groc_amount=abs(round(df.loc[df['Description'].str.contains('\b'+joint_groc_lst+'\b',na=False),
                    'Amount'].sum(),2))
 util_amount=abs(round(df.loc[df['Description'].str.contains('\b'+joint_util_lst+'\b',na=False),
                    'Amount'].sum(),2))

我希望我的函数创建两个变量； groc_amount 和 util_amount，我就可以打印我的结果了。我收到此错误“TypeError: can only concatenate str (not "int") to str" 然后将 str() 添加到我的 for 循环中的 i 并收到以下错误“TypeError: cannot unpack non-iterable int object”

Answer 1

total_joined_list=[joined_groc_lst,joined_util_lst]

groc_amount, util_amount = (abs(round(df.loc[df['Description'].str.contains(f'\b{e}\b',na=False),'Amount'].sum(),2))  for e in total_joined_list)

我已将您的 forloop 更改为一个生成器，该生成器允许解包数据以获取列表中的每个项目。