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

Question

我对 Python 比较陌生，我正在尝试将随机数 (new_bandit) 添加到列表 new_bandit_probs if i == x。但是如果我打印 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)

Answer 1

您的 return 在 for 循环中。在 i 从 0 递增到 1 之前 returns。如果 x!=0 那么它将 return 空列表。

将其移出 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)

Answer 2

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

此外，您的 new_bandit_probs 是空的，因此在第一次迭代后停止，return 一个空列表给出...一个空列表。下面进行一些修改：

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, N_episodes, new_bandit, x):
    for i in range(N_episodes):
        if i == x:
            bandit_probs.append(new_bandit)
            return bandit_probs

print (update_bandit_probs (bandit_probs, N_episodes, new_bandit, x))

输出：

[0.1, 0.4, 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, N_episodes, new_bandit, x))

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

Import elements to a list with for loop

python

list

append