numpyでargsortの逆操作を行う配列を (sort1回のみで) 簡潔に得る

表題のとおりです．

以下のように行うとsortは一回で済み，for文を使う必要もないです．

import numpy as np

# 配列を定義
size = 1 << 20
x = np.random.random(size)

# sortを行う
order = np.argsort(x)

# 逆操作を得る (以下2行)
order_inv = np.zeros_like(order)
order_inv[order] = np.arange(size)

# 検証
np.allclose(x, x[order][order_inv])
>>> True