Python for 循环，从 numpy 排列方法中迭代值

Question

我需要编写代码来测试分类问题的 numpy 截止值数组。要测试的值存储在cutoff_list 变量中。然后，我想将生成的混淆矩阵列表放入字典中。但是，下面的代码只给了我第一个字典条目（第一个测试值的混淆矩阵）：

cutoff_list = [np.arange(0,1,0.01)]  # list of test values
dictionary = {}  
for i, v in enumerate(cutoff_list):
    actual = (df.observed)
    predicted = np.where(df.indicator > i, 1, 0)

    df_confusion = confusion_matrix(actual, predicted) / len(df.indicator)   

    dictionary[i] = df_confusion

print(dictionary)

我正在使用的库：

from pandas import Series, DataFrame
import pandas as pd
import numpy as np
from sklearn.metrics import confusion_matrix

这是循环或字典更新步骤的问题吗？我是 Python 新手，对 R 有更多经验，但仍然在这里苦苦挣扎。任何帮助表示赞赏。

Answer 1

由于您的设置方式，您的 for 循环只运行一次。

您将 np.arrange(0.1.0.01) 的返回值包含在一个列表中，这会破坏您希望 for 循环运行的方式。您只得到一个值，因为 for 循环只运行一次，因为外部列表只有一项。

>>> cutoff_list = [np.arange(0,1,0.01)]
>>> cutoff_list
[array([0.  , 0.01, ... 0.98, 0.99])]
>>> type(cutoff_list)
<class 'list'>

你想得到实际的 numpy 数组：

>>> cutoff_list = np.arange(0,1,0.01)
>>> cutoff_list
array([0.  , 0.01, ... 0.98, 0.99])
>>> type(cutoff_list)
<class 'numpy.ndarray'>

将cutoff_list = [np.arange(0,1,0.01)] 行更改为cutoff_list = np.arange(0,1,0.01)，看看是否能解决您的问题。

我还想你想在这一行中使用v 而不是i：

    predicted = np.where(df.indicator > i, 1, 0)

因为i 将只保存您用作dict 键的枚举值，而v 将保存来自cutoff_list 的值。