python - 索引数组中的两组列

Question

我正在尝试从数组中切出列并分配给一个新变量，就像这样。

array1 = array[:,[0,1,2,3,15,16,17,18,19,20]]

有这样的捷径吗？

我试过了，但它抛出了一个错误:

array1 = array[:,[0:3,15:20]]

这可能真的很简单，但我无法在任何地方找到它。

Answer 1

使用np.r_:

Translates slice objects to concatenation along the first axis.

import numpy as np
arr = np.arange(100).reshape(5, 20)
cols = np.r_[:3, 15:20]

print(arr[:, cols])
[[ 0  1  2 15 16 17 18 19]
 [20 21 22 35 36 37 38 39]
 [40 41 42 55 56 57 58 59]
 [60 61 62 75 76 77 78 79]
 [80 81 82 95 96 97 98 99]]

归根结底，可能只比您现在拥有的内容少一点冗长，但可以在更复杂的情况下派上用场。