Код для обновления каждого элемента (двоичного) массива на основе голосования большинства последних 5 элементов (включая этот элемент)
У меня есть проблема с написанием кода для проблемы, связанной с обновлением элементов массива на основе большинства голосов его последних 5 элементов, включая сам элемент.
Объяснение: Предположим, у нас есть массив с элементами в двоичной форме. arr= [0, 1, 0, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1]
Теперь, согласно нашей задаче, мы должны найти большинство значений последних 5 элементов, включая само число. Таким образом, для первых 4 элементов новое обновленное значение будет равно нулю, а затем значение 4-го индекса будет основано на большинстве голосов прошлых элементов, то есть [0, 1, 0, 0, 0]. Таким образом, значение большинства для индекса 4 равно «0», поскольку количество «0» больше, чем количество «1» в данном окне. Точно так же мы должны сделать то же самое для всех элементов.
ввод обр= [0, 1, 0, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1]
Выходной обр=[0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0]
@AnshumaanMishra для 8-го индекса, у нас есть последние 5 элементов, включая значение в 8-м индексе, равное [0, 1, 0, 1, 1] , поэтому здесь большинство имеет значение «1», поэтому обновленное значение индекса 8 равно «1»
@VishalRaj, тогда вы можете просто проверить, равна ли сумма последних 5 элементов> = 3 (11 в двоичном формате). Если это так, измените его на 1, иначе на 0.
Ответы 1
Использовать:
arr = [0 , 1, 0, 0, 0, 1, 0, 1 , 1, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1]
out = [int(sum(arr[max(i-5, 0):i]) >= 3) for i in range(1, len(arr)+1)]
print(out)
# Output
[0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0]
Вариация:
out = [int(sum(arr[max(i-4, 0):i+1]) >= 3) for i in range(len(arr))]
Почему индекс 8
1потому что до индекса 8 5 нулей и 4 единицы. Разве не должно быть0?