2024-04-19 21:28:38 发布
网友
我有一个2dnumpy数组,A包含另一个数组的索引B。使用numpy从A和{}获取{}的好方法是什么?
A
B
A = array([[1, 1, 0, 2], [1, 0, 0, 2], [1, 1, 0, 2]]) B = array([0, 5, 3]) C = array([[5, 5, 0, 3], [5, 0, 0, 3], [5, 5, 0, 3]])
这个怎么样C = B[A]。这就是纽比的美丽之处:
C = B[A]
In [1]: import numpy as np In [2]: A = np.array([[1, 1, 0, 2], ...: [1, 0, 0, 2], ...: [1, 1, 0, 2]]) In [3]: B = np.array([0, 5, 3]) In [4]: B[A] Out[4]: array([[5, 5, 0, 3], [5, 0, 0, 3], [5, 5, 0, 3]])
x,y = a.shape for i in range(x): for j in range(y) c[i][j] = b[a[i][j]]
这个怎么样
C = B[A]
。这就是纽比的美丽之处:相关问题 更多 >
编程相关推荐