python - 根据 num 条件返回 nums 列表

Question

如果列表中的 num 大于 0.5，我想得到 1，否则得到 0。

例如，a是一个列表

Matlab:

b = (a > 0.5)

Python:

b = [1 if x > 0.5 else 0 for x in a]
b = [int(x > 0.5) for x in a]

我想知道Python中是否有更短的表达式？

Answer 1

如果 a 是矩阵，则最好使用 numpy 数组。这基本上是 Python 中的 matlab 对应部分:

>>> import numpy as np
>>> a = np.array([0.35,0.76,0.22,0.10])
>>> a > 0.5
array([False,  True, False, False], dtype=bool)
>>> (a > 0.5).astype(int)
array([0, 1, 0, 0])

因此，a > 0.5 构造了一个与原始数组具有相同维度的数组，并使用存储 True 的 bool 值，以防数字大于 0.5 code>，否则 False。

通过使用.astype(int)，您可以将其转换为其他类型( bool 值除外)，例如int。

Numpy 可以处理任意维度的数组，而且调用是由 Python 外部的专用包完成的，从而使其性能更好。