如何将其变成列表理解？

问题描述

我有一个字符串列表。其中一些字符串可以转换为浮点数，其他则不能。我正在尝试提取可以转换为浮点数的那些。现在，我使用以下代码

my_strings = ['1','42','.3','a','b','c']
my_floats = []
for e in my_strings:
    try:
        my_floats.append(float(e))
    except ValueError:
        pass

有没有办法把这个操作作为列表推导式来做？

解决方法

您可以编写一个 safeFloat 函数，将字符串转换为浮点数，并对无效字符串使用默认值。如果您不想要不是数字的项目，这可以用于列表理解或使用过滤器映射：

def safeFloat(S,default=None):
    try:    return float(S)
    except: return default


my_strings = ['1','42','.3','a','b','c']

my_floats  = [safeFloat(s,0) for s in my_strings]

[1.0,42.0,0.3,0]

my_floats = [float(s) for s in my_strings if safeFloat(s) != None]

[1.0,0.3]

my_floats  = list(filter(None,map(safeFloat,my_strings)))

[1.0,0.3]

如果您经常在列表或迭代器上执行此操作，您可以进一步将转换封装在一个函数中，用于转换和过滤多个项目

def getFloats(S,default=None):
    yield from filter(None,S))

my_floats = [*getFloats(my_strings)]

[1.0,0.3]

,

您的代码逻辑不应依赖异常。试试这个：

my_strings = ['1','c']
my_floats = [float(string) for string in my_strings if string.replace('.','',1).isdigit()]