Заменить элементы в списке списков в соответствии с индексами, хранящимися в другом списке

У меня есть следующие массивы:

example_array = [[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0]]
indexes = [[1,2,3], [2,3], [0,4,2], [1,4,3], [0,4,3], [1,2,3], [1,2]]

Я хотел бы заменить элементы в example_array на 1 в соответствии с индексом в indexes. Итак, ожидаемый результат должен быть таким:

example_array = [[0,1,1,1,0],[0,0,1,1,0],[1,0,1,0,1],[0,1,0,1,1],[1,0,0,1,1],[0,1,1,1,0],[0,1,1,0,0]].

Я пробовал разные подходы, например, использовать панд, например:

matrix = pd.DataFrame(example_array)

for row in matrix:
    for i in indexes:
        for j in i:
            x_u.iloc[x, j] = 1

но это не дает мне ожидаемых результатов! Решение не требует использования библиотеки pandas. Спасибо за помощь!

python numpy matrix indexing

27.05.2018 13:49

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

167

Перейти к ответу Данный вопрос помечен как решенный

Ответы 6

Ответ принят как подходящий

Вы можете просто сделать это:

example_array = [[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0]]
indexes = [[1,2,3], [2,3], [0,4,2], [1,4,3], [0,4,3], [1,2,3], [1,2]]

for i,index in enumerate(indexes):
  for idx in index:
    example_array[i][idx] = 1
print example_array

Выход:

[[0, 1, 1, 1, 0], [0, 0, 1, 1, 0], [1, 0, 1, 0, 1], [0, 1, 0, 1, 1], [1, 0, 0, 1, 1], [0, 1, 1, 1, 0], [0, 1, 1, 0, 0]]

27.05.2018 13:54

Попробуй это:

example_array = [[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0]]
matrix = pd.DataFrame(example_array)
indexes = [[1,2,3], [2,3], [0,4,2], [1,4,3], [0,4,3], [1,2,3], [1,2]]
for row_num in range(len(matrix)):
    matrix.iloc[row_num][indexes[row_num]] = 1

Это также избавляет вас от вложенного цикла!

27.05.2018 13:57

Используя массив numpy, вы можете напрямую установить указанные индексы на желаемое значение.

import numpy as np
example_array = np.array([[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0]])
indexes = [[1,2,3], [2,3], [0,4,2], [1,4,3], [0,4,3], [1,2,3], [1,2]]

for i, ind in enumerate(indexes):
    example_array[i, ind] = 1
print(a)

Выход

[[0 1 1 1 0]
 [0 0 1 1 0]
 [1 0 1 0 1]
 [0 1 0 1 1]
 [1 0 0 1 1]
 [0 1 1 1 0]
 [0 1 1 0 0]]

27.05.2018 13:59

example_array = [[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0],[0,0,0,0,0]];
indexes = [[1,2,3], [2,3], [0,4,2], [1,4,3], [0,4,3], [1,2,3], [1,2]];

for i in range(0, len(example_array)):
    for j in range(0, len(indexes[i])):
        example_array[i][indexes[i][j]] = 1;

print example_array;

27.05.2018 14:05

Или вы можете использовать понимание:

result = [[1 if i in ind else e for i, e in enumerate(ea)] for ea, ind in zip(example_array, indexes)]

27.05.2018 14:09

с одним вкладышем:

indices = [[1, 2, 3], [2, 3], [0, 4, 2], [1, 4, 3], [0, 4, 3], [1, 2, 3], [1, 2]]
print([[1 if idx in sub else 0 for idx in range(5)] for sub in indices])

выход:

[[0, 1, 1, 1, 0], [0, 0, 1, 1, 0], [1, 0, 1, 0, 1], [0, 1, 0, 1, 1], [1, 0, 0, 1, 1], [0, 1, 1, 1, 0], [0, 1, 1, 0, 0]]

27.05.2018 14:09

Другие вопросы по теме

Массив фрагментов Python при условии

Извлечение стен из изображения в OpenCv

Получить список имен столбцов, все значения - NaN в Python

Логическое значение нескольких условий массива Numpy

Python: более быстрая альтернатива numpy random.choice ()?

Какова цель и полезность параметра subok в numpy.zeros_like ()?

Заполнение столбцов 2D массива по 1D массиву счетчиков

Учебник Tensorflow по MNIST

Python numpy: linalg.pinv () слишком неточный

Numpy: заполнение массивов для векторизации вложенных сумм

Заменить элементы в списке списков в соответствии с индексами, хранящимися в другом списке

Ответы 6

Другие вопросы по теме

Похожие вопросы