代码之家 › 专栏 › 技术社区 › palazzo train

如何通过另一个数组中的索引更新ndarray中的值[duplicate]

numpy-ndarray multidimensional-array numpy python

palazzo train · 技术社区 · 6 年前

例如,我有10x7个零数组 x=np.zeros( (10,7) )

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.],
       [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.]])

我想在每一行中随机分配一个'1'。假设我创建了另一个数组(10,1),然后值在0-6之间。 r=np.random.randint(0, 7, (10,1))

array([[6],
       [2],
       [5],
       [1],
       [2],
       [4],
       [6],
       [3],
       [0],
       [1]])

我想从 r 这意味着将元素x[0,6],x[1,2],x[2,5],x[3,1]等设为0,那么x应该变成

array([[0., 0., 0., 0., 0., 0., 1.],
       [0., 0., 1., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 1., 0.],
       [0., 1., 0., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0., 0.],
       [0., 0., 0., 0., 1., 0., 0.],
       [0., 0., 0., 0., 0., 0., 1.],
       [0., 0., 0., 1., 0., 0., 0.],
       [1., 0., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0., 0.]])

如何有效地做到这一点?

1 回复 | 直到 6 年前

Mazdak 6 年前

将一维数组用于 r 并将其用作列索引。对于行索引,您只需使用一个范围:

In [25]: r=np.random.randint(0, 7, 10)

In [26]: x=np.zeros( (10,7) )

In [27]: x[np.arange(10), r] = 1

In [28]: x
Out[28]: 
array([[0., 1., 0., 0., 0., 0., 0.],
       [0., 0., 0., 0., 0., 0., 1.],
       [0., 0., 0., 0., 0., 0., 1.],
       [0., 1., 0., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0., 0.],
       [0., 0., 0., 1., 0., 0., 0.],
       [0., 0., 0., 0., 1., 0., 0.],
       [0., 0., 1., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0., 0.]])

推荐文章

Paul Williams · 迭代数组时输出有问题

8 月前

pasty guacamole · 在结构体(C)中使用双指针访问2d数组上的信息

1 年前

Muffinman · 使用Excel O365,如何在两列上使用唯一性,对相关数据进行排序和复制?

1 年前

Suyash Nachankar · 为什么矩阵的所有行都与最后一行相同?

1 年前

Vyacheslav Verkhovin · C中二维数组的Typedef和指针上的访问

1 年前

Aditi · 我正在尝试用Java解决一个数字模式问题。我已经写了代码,如果有人能对我现有的sol进行更改,我将不胜感激

1 年前

neo · 为什么我不能访问以取消引用int指针作为索引的char指针数组中的位置

1 年前

Omprakash · 使用Google工作表中的多行和多列进行筛选

1 年前

Genki · Bash:如何从多维数组中的命名键获取值

1 年前

Insert0Name0Here · 将多维数组中的数组传递给C中的函数#

1 年前