numpy中是否有一种方法可以检索数组中除提供的索引项之外的所有项.
x =
array([[[4, 2, 3],
[2, 0, 1],
[1, 3, 4]],
[[2, 1, 2],
[3, 2, 3],
[3, 4, 2]],
[[2, 4, 1],
[0, 2, 2],
[4, 0, 0]]])
Run Code Online (Sandbox Code Playgroud)
并要求
x[not 1,:,:]
Run Code Online (Sandbox Code Playgroud)
你会得到
array([[[4, 2, 3],
[2, 0, 1],
[1, 3, 4]],
[[2, 4, 1],
[0, 2, 2],
[4, 0, 0]]])
Run Code Online (Sandbox Code Playgroud)
谢谢
In [42]: x[np.arange(x.shape[0])!=1,:,:]
Out[42]:
array([[[4, 2, 3],
[2, 0, 1],
[1, 3, 4]],
[[2, 4, 1],
[0, 2, 2],
[4, 0, 0]]])
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
336 次 |
| 最近记录: |