假设我有一个 NumPy
的二维数组 A
:
>>> import numpy as np
>>> A=np.arange(30).reshape(3,10)
>>> A
array([[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19],
[20, 21, 22, 23, 24, 25, 26, 27, 28, 29]])
我需要获取两个数组 B
和 C
,这两个数组需要具备以下特性:
B = array([[ 0, 3, 4, 5, 6, 7, 8, 9],
[10, 13, 14, 15, 16, 17, 18, 19],
[20, 23, 24, 25, 26, 27, 28, 29]])
C = array([[ 1, 2],
[11, 12],
[21, 22]])
如何最简单地完成这个任务?
请注意,我必须获取所有C
(2个相邻列)和B
(即没有C
的A
)的集合。我尝试了不同的NumPy
构造,例如np.delete
,np.hstack
,但在上面的示例中似乎都无法处理边缘条件。