python - 列表推导 if condition else 替换

Question

我正在尝试用高度值格式化列表，但有些元素不是值

hei=['NN', '99.0m', '99.0m', '99.0m', '99.0m', '97.0m', '97.0m', '97.0m','rat_v2']

所以我想用 rstrip 删除 'm' 并替换 '_' 的非数值

这是我的尝试

heiFormat=[k.rstrip('m') if k[0].isdigit else k.replace(k,'_') for k in hei]

但结果并不完全正确，好像没有替换非数字

heiFormat=['NN', '99.0', '99.0', '99.0', '99.0', '97.0', '97.0', '97.0','rat_v2']

虽然我想要类似的东西

heiFormat=['_', '99.0', '99.0', '99.0', '99.0', '97.0', '97.0', '97.0','_']

有什么问题或缺失？

Answer 1

当你做 - k[0].isdigit - 您只是访问该函数，而不是调用它，这就是该条件对所有值的计算结果为真的原因。

你应该调用它。而且你也不需要 replace ，更简单的方法是直接输入 '_'有k.replace(k,'<something>')和直接用'<something>'是一样的.

例子-

>>> heiFormat=[k.rstrip('m') if k[0].isdigit() else '_' for k in hei]
>>> heiFormat
['_', '99.0', '99.0', '99.0', '99.0', '97.0', '97.0', '97.0', '_']