如何按特定顺序重新排列我的 Numpy 数组？

Question

我想重新排列以下 Numpy 数组：

X = [ 1.  5.  2.  4.  2.  4.  1.  5.  2.  1.  2.  1.  5.  6.  2.  6.  5.  4.
  3.  1.  4.  6.  5.  3.  1.  5.  4.  5.  3.  3.  1.  4.  4.  5.  4.  4.
  3.  6.  1.  5.  4.  1.  4.  4.  1.  5.  1.  2.  1.  4.  6.  1.  3.  4.
  1.  6.  3.  1.  1.  5.  6.  4.  5.  2.  6.  3.  1.  3.  4.  6.  3.  2.
  1.  4.  2.  4.  2.  1.  2.  2.  1.  1.  6.  4.  3.  6.  1.  1.  4.  1.
  4.  4. nan nan nan nan]

按以下顺序3, 2, 6, 5, 4, 1。

本质上，整个数组的排列方式是所有的 3 在最前面，2，6，5，4，最后是 1。在保留 Numpy 数组而不是将其转换为列表的同时，最好的方法是什么？

Answer 1

使用list.index：

ind = [3, 2, 6, 5, 4, 1]
sorted(X, key=lambda x : ind.index(x) if x in ind else -1)

输出：

[nan,
 nan,
 nan,
 nan,
 3,
 ...
 1,
 1,
 1]

如果你希望nan 最后出现：

last = len(ind)
sorted(X, key=lambda x : ind.index(x) if x in ind else last)

输出：

[3,
 3,
 ...
 1,
 1,
 nan,
 nan,
 nan,
 nan]