将列表中的浮点数近似为标量整数时抛出错误

Question

我试图将列表中的浮点数近似为标量整数值。也就是说，我试图将小于 0.5 的值转换为 0，并将大于或等于 0.5 的值转换为 1。但我收到错误。我的列表值如下所示：0.2943,0.3483,0.3359,0.3671,0.6788,1,0.779预期输出：0,0,0,0,1,1,1编写的代码：listSample = []listSample = list(y_predAN_PCA)for i in listSample:    if listSample[i] < 0.5:        listSample[i] = 0    else:        listSample[i] = 1但我收到以下错误：类型错误：只有整数标量数组可以转换为标量索引

Answer 1

你已经得到了别人的答案。如果您使用第三方库没有问题，您可以使用 NumPy 数组进行掩码。对于这样一个小的示例案例，这将是一种矫枉过正，但了解选项仍然很好。

说明： lst<0.5返回值小于 0.5 的数组索引。然后将其作为索引传递给数组lst，lst[lst<0.5]并重新分配这些值0。类似地，您检查大于等于 0.5 的值并将它们重新分配为 1。

import numpy as np

lst = np.array([0.2943,0.3483,0.3359,0.3671,0.6788,1,0.779])

lst[lst<0.5] = 0

lst[lst>=0.5] = 1

print (lst)

# array([0., 0., 0., 0., 1., 1., 1.])

Jon Clements建议的另一种更好的方法是使用np.where。在这里，您首先指定条件（lst<0.5此处）。如果条件为True，则条件后的第一个值将分配给数组元素。如果条件为False，则将分配第二个值。

np.where(lst<0.5, 0, 1)

# array([0., 0., 0., 0., 1., 1., 1.])

Answer 2

您在代码片段中错误地使用了 for 循环。在您的代码中，我指的是列表中的第i个值，而不是第i个索引。下面的代码可以提供帮助。

range 函数将根据列表的长度生成索引。

for i in range(len(listSample)):

    if listSample[i] < 0.5:

        listSample[i] = 0

    else:

        listSample[i] = 1

Answer 3

您正在迭代项目而不是项目索引；用于enumerate获取两者：

listSample = list(y_predAN_PCA)

for i,s in enumerate(listSample):

    if s < 0.5:

        listSample[i] = 0

    else:

        listSample[i] = 1

更好的是，使用round和列表理解：

listSample = [round(e) for e in listSample]

或者，如果您使用的是 NumPy：

listSample = np.round(listSample)

注意：这会将 0.5向下舍入为 0（“银行家的舍入”）并返回浮点数列表。更接近您的代码（对于正数）将是：

listSample = [int(e+0.5) for e in listSample]