python - 如何根据索引值生成列表

Question

我正在尝试根据用户提示生成一个由 0 组成的列表。如果他们输入 1，则 1 索引将被 1 替换，如果他们输入 2，则 2 索引将被 1 替换，等等。我能够生成 1 和 0 的随机列表，但我遇到了麻烦与输入。
这是我到目前为止所拥有的:

import random

def askTheUser():
  number = input("Do you want to roll again? Pick a number or numbers thru 0 and 5:")
  myList = []
  aList = [1,0]
  for i in range(5):
    myList.append(random.choice(aList))
    if number == 1:
        return myList[1] = 0
    if number == 2:
        return myList[2] = 0

return myList




print(askTheUser())

Answer 1

我认为您正在替换为 0 而不是 1，输入也采用字符串而不是 int，所以尝试对其进行强制转换 并且列表索引是从 0 开始的，所以正确的代码应该是:

import random

def askTheUser():
  number = input("Do you want to roll again? Pick a number or numbers thru 0 and 4:")
  myList = []
  aList = [1,0]
  for i in range(5):
    myList.append(random.choice(aList))
  myList[int(number)] = 1

  return myList
print(askTheUser())