使用冒号符号在NumPy数组中选择特定列

Question

5

我有一个40000行60列的Numpy数组，我想要做类似这样的操作：

mat[:,[0:13,19:23,23:31,39:59]]

显然，这种方法行不通。有没有比连接更聪明的方法？

- Gordon Cai

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Divakar · Accepted Answer

使用 np.r_ 函数 -

mat[:,np.r_[0:13,19:23,23:31,39:59]]

示例运行 -

In [48]: mat = np.random.rand(100,1000)

In [50]: mat[:,np.r_[0:13,19:23,23:31,39:59]].shape
Out[50]: (100, 45)

In [51]: 13+4+8+20
Out[51]: 45