我如何使用numpy从另一个阵列的列（维数为2 * 4）形成一个二维数组给定一个列数有效索引数组

Question

我想通过n使用numpy，内部元素来自列号数组所选的特定列。 例如，如果我有这样的事情

[[1, 2, 3], 
[2, 3, 4]] 

我的输入数组，我想有列

[2,3,1,2,3]，

我会得到

[[2, 3, 1, 2, 3], 
    [3, 4, 2, 3, 4]] 

我的输出阵列

Answer 1

您想沿第二 尺寸。但是请记住，numpy使用基于零的索引。你需要的[1, 2, 0, 1, 2]代替[2, 3, 1, 2, 3]

a = np.array([ 
    [1, 2, 3], 
    [2, 3, 4]]) 

a[:, [1, 2, 0, 1, 2]] 

array([[2, 3, 1, 2, 3], 
     [3, 4, 2, 3, 4]]) 

​