Question

我正在做并行化工作。我之间有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那样的.. 给我一个建议..

Answer 1

如果你想引用数组的索引2（即最右边一列）的列，你可以这样做

pos[:,2]

如果要按第二列对pos数组进行排序（就地），可以执行以下操作：

pos[:,2].sort()

Numpy数组1中的元素索引

1 个答案: