您可以通过在numpy中使用高级索引来实现这一点：

A[np.arange(len(B)), B] += 1

其工作原理是使用np.arange(len(B))（表示行索引）创建维度为(len(B), len(B))的2D数组。高级索引的第二个索引B表示列索引。通过将A[np.arange(len(B)), B]加1，可以增加B指定的每行中的元素。

在numpy中，您可以通过使用数组的arrange和shape来完成

import numpy as np

A = np.array([[0, 1, 3, 5, 9], 
              [2, 7, 5, 1, 4]])
B = np.array([2, 4])

A[np.arange(A.shape[0]), B] += 1

print(A)

np.arange(A.shape[0])从0 to A.shape[0] - 1生成整数数组。A.shape[0]基本上是行
你也可以用循环来做。

import numpy as np
A = np.array([[0, 1, 3, 5, 9], 
              [2, 7, 5, 1, 4]])
B = np.array([2, 4])

for i, index in enumerate(B):
    A[i][index] += 1

print(A)