A = np.asarray([[0,1,2],[3,4,5],[3,4,5],[6,7,8]])
a = np.asarray([3,4,5])
print(A)
array([[0, 1, 2],
[3, 4, 5],
[3, 4, 5],
[6, 7, 8]])
在A中找所有a所在的行:
idx = np.argwhere((A == a).all(axis=1)).ravel()
print(idx)
[1 2]
A = np.asarray([[0,1,2],[3,4,5],[3,4,5],[6,7,8]])
a = np.asarray([3,4,5])
print(A)
array([[0, 1, 2],
[3, 4, 5],
[3, 4, 5],
[6, 7, 8]])
在A中找所有a所在的行:
idx = np.argwhere((A == a).all(axis=1)).ravel()
print(idx)
[1 2]