Python 列表推导根据字符串匹配修改元素

Question

我有 python 代码，它返回列表理解，如下面的返回代码所示。

return [str(source.get(field)) for field in fields]
# one of the list output
['3', 'checkuser', '=checkuser', 'management', '', 'checkuser@gmail.com', '', 'True', "['apiusermanagement']", 'False']

要求

我需要检查列表中的每个元素，如果元素的第一个字符以 ('=', '@', '|', '%') 这些字符中的任何一个开头，那么我需要 append 中的单双引号 (")该列表元素的开始。

例如，示例 output 中的第 3 个元素 '=checkuser' 以 '=' 开头，因此该元素应修改为 '"=checkuser' 并且应该是返回列表 output 的一部分

# Expected output
['3', 'checkuser', '"=checkuser', 'management', '', 'checkuser@gmail.com', '', 'True', "['apiusermanagement']", 'False']

我是 Python 的初学者，并尝试以最小的更改和良好的性能 output 来实现这一目标。 任何帮助或指针将不胜感激

Answer 1

一种选择是将此逻辑放入 function 中：

def fetch_and_map(field):
    result = str(source.get(field))
    if result.startswith("="):
        return '"' + result
    else:
        return result

return [fetch_and_map(field) for field in fields]

你也可以这样内联：

return ['"' + str(source.get(field)) if str(source.get(field)).startswith("=") else str(source.get(field)) for field in fields]

请注意，如果source是字符串的字典，则可以通过跳过str()调用来简化：

return ['"' + source.get(field) if source.get(field).startswith("=") else source.get(field) for field in fields]

Answer 2

您可以使用以下代码：

output = ['3', 'checkuser', '=checkuser', 'management', '', 'checkuser@gmail.com', '', 'True', "['apiusermanagement']", 'False']
output_modified = []
for string in output:
    if string != "" and string[0] in ['=', '@', '|', '%']:
        output_modified.append('"' + string)
    else:
        output_modified.append(string)

这将遍历output中的每个字符串。 如果字符串不为空且以'=' 、 '@' 、 '|'开头 , 或'%' ，它将在string的开头添加'"'并将 append 添加到output_modified 。如果没有，它会简单地将 append 原样添加到output_modified 。

Answer 3

将列表放入变量中。 然后循环变量，在需要的地方添加双引号。

result = [str(source.get(field)) for field in fields]
for i, val in enumerate(result):
    if len(val) > 0 and val[0] in ('=', '@', '|', '%'):
        result[i] = '"' + val
return result

Answer 4

像这样的东西

lst = ['3', 'checkuser', '=checkuser', 'management', '', 'checkuser@gmail.com', '', 'True', "['apiusermanagement']",
       'False']
lst_ex = [f'"{x}' if x and x[0] in {'=', '@', '|', '%'} else x for x in lst]
print(lst_ex)

output

['3', 'checkuser', '"=checkuser', 'management', '', 'checkuser@gmail.com', '', 'True', "['apiusermanagement']", 'False']