例如数组:
array = np.array([[1, 1, 4, 2, 1, 8],
[1, 1, 8, 2, 1, 16],
[1, 1, 40, 2, 1, 80],
[1, 2, 40, 2, 1, 80]])
我想基本上go 掉内部数组的每三个[:, ::2]
个元素.所以结果应该是:
[[1 1 2 1]
[1 1 2 1]
[1 1 2 1]
[1 2 2 1]]
我可以做两个 Select 并连接每个结果数组,但这似乎非常慢.有没有一种方法可以使用索引或其他方法,而不涉及np.linatenate的循环?
我的实际数组总是有大小可以被3整除的内部数组,而1维非常非常大.因此,我感兴趣的是实现这一目标的最快方法.
谢谢!