根据存储在另一个列表中的索引替换列表列表中的元素答案

【问题标题】：Replace the elements in list of lists according to the indexes stored in another list根据存储在另一个列表中的索引替换列表列表中的元素
【发布时间】：2018-11-06 04:32:06
【问题描述】：

我有以下数组：

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。所以，预期的结果应该是这样的：

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]].

我尝试了不同的方法，例如使用 pandas，如下所示：

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

【解决方案1】：

你可以这样做：

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]]

【讨论】：

【解决方案2】：

试试这个：

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

它还为您节省了嵌套循环！

【讨论】：

【解决方案3】：

使用 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]]

【讨论】：

【解决方案4】：

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;

【讨论】：

【解决方案5】：

或者你可以使用推导：

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

【讨论】：

【解决方案6】：

只有一个衬里：

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]]

【讨论】：