如何将其变成列表理解?
How to make this into a list comprehension?
我有一个字符串列表。其中一些字符串可以转换为浮点数,其他则不能。我正在尝试提取那些可以转换为浮点数的。现在,我正在使用以下代码
my_strings = ['1', '42', '.3', 'a', 'b', 'c']
my_floats = []
for e in my_strings:
try:
my_floats.append(float(e))
except ValueError:
pass
有没有办法将此操作作为列表理解来执行?
您的代码逻辑不应依赖异常。试试这个:
my_strings = ['1', '42', '.3', 'a', 'b', 'c']
my_floats = [float(string) for string in my_strings if string.replace('.','',1).isdigit()]
您可以编写一个 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, 0, 0]
my_floats = [float(s) for s in my_strings if safeFloat(s) != None]
[1.0, 42.0, 0.3]
my_floats = list(filter(None,map(safeFloat,my_strings)))
[1.0, 42.0, 0.3]
如果您经常在列表或迭代器上这样做,您可以进一步将转换封装在一个函数中,该函数可以转换和过滤多个项目
def getFloats(S,default=None):
yield from filter(None,map(safeFloat,S))
my_floats = [*getFloats(my_strings)]
[1.0, 42.0, 0.3]
我有一个字符串列表。其中一些字符串可以转换为浮点数,其他则不能。我正在尝试提取那些可以转换为浮点数的。现在,我正在使用以下代码
my_strings = ['1', '42', '.3', 'a', 'b', 'c']
my_floats = []
for e in my_strings:
try:
my_floats.append(float(e))
except ValueError:
pass
有没有办法将此操作作为列表理解来执行?
您的代码逻辑不应依赖异常。试试这个:
my_strings = ['1', '42', '.3', 'a', 'b', 'c']
my_floats = [float(string) for string in my_strings if string.replace('.','',1).isdigit()]
您可以编写一个 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, 0, 0]
my_floats = [float(s) for s in my_strings if safeFloat(s) != None]
[1.0, 42.0, 0.3]
my_floats = list(filter(None,map(safeFloat,my_strings)))
[1.0, 42.0, 0.3]
如果您经常在列表或迭代器上这样做,您可以进一步将转换封装在一个函数中,该函数可以转换和过滤多个项目
def getFloats(S,default=None):
yield from filter(None,map(safeFloat,S))
my_floats = [*getFloats(my_strings)]
[1.0, 42.0, 0.3]