Numpy数组1中的元素索引

时间:2014-09-01 13:26:12

标签: python numpy multidimensional-array

我正在做并行化工作。我之间有numpy数组。 我想检查numpy数组的索引 例如:

pos是一个ndarray,它的值是

 pos :

 ndarray[[    0.    44.  2367.]
         [    0.    73.  2301.]
         [    0.    38.  2304.]
         [    0.    35.  2349.]
         [    0.     6.     0.]
         [    0.    43.  2319.]
         [    0.    95.  2381.]
         [    0.   189.     3.]
         [    0.     0.  2339.]
         [    0.    22.  2335.]
         [    0.    44.  2345.]
         [    0.    44.  2345.]
         [    0.    52.  2348.]
         [    0.    50.  2348.]]

 dtype :float64

 max : 2381.0
 min :0.0

 Shape = {tuple}(14, 3)

 size :42

我想在我的计算中只考虑pos ndarray的第二个位置,例如2367,2301,2304 ...等等。 那么我如何索引它们以及如何才能对第二个位置值进行排序?意味着如果数组的第二个位置是3,4,2,1那么我想第二个位置是1,2,3,4那样的.. 给我一个建议..

1 个答案:

答案 0 :(得分:1)

如果你想引用数组的索引2(即最右边一列)的列,你可以这样做

pos[:,2]

如果要按第二列对pos数组进行排序(就地),可以执行以下操作:

pos[:,2].sort()