【Numpy】np.add.at 示例

二进制人工智能

于 2024-04-29 14:09:21 发布

阅读量482

点赞数 2

分类专栏： python 文章标签： numpy

本文链接：https://blog.csdn.net/weixin_44378835/article/details/138310829

版权

python 专栏收录该内容

23 篇文章

订阅专栏

文章介绍了如何使用NumPy库中的`np.add.at`函数，在一维和二维数组中根据给定的索引位置进行元素级别的加法操作，以实现特定元素值的累加。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一维数组

import numpy as np

n_classes = 4
vector = np.zeros([n_classes])
print(vector)
idx = np.array([1, 3, 0])

# 在[1], [3], [0] 位置上加1
np.add.at(vector, idx , 1)
print(vector)

[0. 0. 0. 0.]

[1. 1. 0. 1.]

二维数组

import numpy as np

n_classes = 4
confusion_matrix = np.zeros([n_classes, n_classes])
print(confusion_matrix)
idx = np.array([1, 3, 0])
idy = np.array([1, 2, 1])

# 在[1,1]、[3,2]、[0,1] 位置上加1
np.add.at(confusion_matrix, (idx , idy), 1)
print(confusion_matrix)

[[0. 0. 0. 0.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]]

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