使用 for 循环将元素导入列表

如何解决使用 for 循环将元素导入列表

我对 Python 比较陌生，如果 i == x，我正在尝试向列表 new_bandit_probs 添加一个随机数 (new_bandit)。但是如果我打印 new_bandit_probs 列表，我只会得到一个空列表。有人可以帮我吗？

import numpy as np

bandit_probs = [0.1,0.4,0.9] 
N_bandits = len(bandit_probs)
N_experiments = 1 
N_episodes = 100 
new_bandit = np.random.random ()
x = np.random.randint(0,N_episodes)
new_bandit_probs = []

def update_bandit_probs (bandit_probs,N_episodes,new_bandit,x,new_bandit_probs):
    for i in range(N_episodes):
        if i == x:
            new_bandit_probs.append(bandit_probs,new_bandit)
        return(new_bandit_probs)

解决方法

您的返回在 for 循环内。它在 i 从 0 增加到 1 之前立即返回。如果 x!=0 那么它将返回空列表。

将其移出 for 循环：

def update_bandit_probs (bandit_probs,N_episodes,new_bandit,x,new_bandit_probs):
    for i in range(N_episodes):
        if i == x:
            new_bandit_probs.append(bandit_probs,new_bandit)
    return(new_bandit_probs)

,

您的代码无法正常工作，因为您的 return 是在 for 循环中的第一次迭代之后设置的。

另外，你的 new_bandit_probs 是空的，所以在第一次迭代后停止并返回一个空列表给出......一个空列表。下面进行一些修改：

import numpy as np

bandit_probs = [0.1,0.4,0.9] 
N_bandits = len(bandit_probs)
N_experiments = 1 
N_episodes = 100 
new_bandit = np.random.random ()
x = np.random.randint(0,N_episodes)

def update_bandit_probs (bandit_probs,x):
    for i in range(N_episodes):
        if i == x:
            bandit_probs.append(new_bandit)
            return bandit_probs

print (update_bandit_probs (bandit_probs,x))

输出：

[0.1,0.9,0.4026139866070344]

您也可以在一行中执行此操作：

def update_bandit_probs(v,n,c,x):
    return v+[c for i in range(n) if x == i]

print (update_bandit_probs(bandit_probs,x))