如何按索引比較 python 中的兩個列表

Question

我必須按索引比較兩個列表，如果列表 1 中的索引等於 0，則列表 2 中的所有元素都為 0。

list1 = [0, 1, 1, 0]   # that's for example 
list2 = [[15, 19, 13, 15, 30, 14, 14], [14, 22, 30, 19, 29, 17, 15], [19, 21, 11, 25, 23, 23, 30], [15, 15, 25, 18, 22, 24, 29], [24, 30, 30, 11, 25, 18, 27]]

output：

list2 = [[0, 0, 0, 0, 0, 0, 0], [14, 22, 30, 19, 29, 17, 15], [19, 21, 11, 25, 23, 23, 30], [0, 0, 0, 0, 0, 0, 0]]

Answer 1

list1 = [0, 1, 1, 0] 
list2 = [[15, 19, 13, 15, 30, 14, 14], [14, 22, 30, 19, 29, 17, 15], [19, 21, 11, 25, 23, 23, 30], [15, 15, 25, 18, 22, 24, 29], [24, 30, 30, 11, 25, 18, 27]]
new_list=[]
for i in list1:
  for index,j in enumerate(list2):
    if(i==index):
      new_list.append([x*i for x in j] )
      break 
print(new_list)

Answer 2

這也應該這樣做：

list1 = [0, 1, 1, 0]   # that's for example 
list2 = [[15, 19, 13, 15, 30, 14, 14], [14, 22, 30, 19, 29, 17, 15], [19, 21, 11, 25, 23, 23, 30], [15, 15, 25, 18, 22, 24, 29], [24, 30, 30, 11, 25, 18, 27]]

result = [[0] * len(list2[index])  if index == 0 else list2[index] for index in list1]

print(result)

Answer 3

我的看法沒有任何 if 語句

list1 = [0, 1, 1, 0]
list2 = [[15, 19, 13, 15, 30, 14, 14], [14, 22, 30, 19, 29, 17, 15], [19, 21, 11, 25, 23, 23, 30], [15, 15, 25, 18, 22, 24, 29], [24, 30, 30, 11, 25, 18, 27]]

output_list = []
for index, item in enumerate(list1):
    output_list.append([element * item for element in list2[index]])
print(output_list)