python - 根据大于一行的值分配 0 和 1

Question

假设我有一个包含 1 到 10 整数的数组，必须用 0 替换所有小于 6 的整数，用 1 替换所有等于或大于 6 的整数。目前，我正在这样做:

arry[arry < 6] = 0
arry[arry >= 6] = 1

我想知道有什么方法可以将这两个语句组合成一行代码，或者任何其他解决方案。

Answer 1

我假设 arry 是一个 numpy 数组(您使用的智能索引似乎表明了这一点)。在这种情况下，您可以简单地执行以下操作:

arry = (arry >= 6).astype(int)

其中 astype(int) 会将 bool 数组 arry >= 6 转换为整数数组。