首页 > 解决方案 > 使用另一个数组创建一个由 1 和 0 组成的数组

问题描述

所以我有两个数组(行和列)row = [0, 1, 0, 2, 0, 1, 3, 1, 3, 1, 2, 4]column = [0, 0, 1, 1, 2, 2, 2, 3, 3, 0, 1, 4]

我想使用行和列数组将值“1”插入另一个二维数组(向量)

vectors = 
[[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.]]

所以我想要的输出是:

vectors = 
[[1. 1. 1. 0. 0.],
[1. 0. 1. 1. 0.],
[0. 1. 0. 0. 0.],
[0. 0. 1. 1. 0.],
[0. 0. 0. 0. 1.]]

抱歉,如果我的解释不好,这是我第一次使用 python 和 Stackoverflow。

标签: pythonarraysmultidimensional-array

解决方案


我想你可以试试这个:

for r, c in zip(row, column):
    vectors[r][c] = 1

推荐阅读