我想轉換這段代碼以使其與 Numba 兼容。Numba 支持的唯一排序方法是sorted()但不使用鍵arg。我必須在沒有其他庫匯入或者只是一些 numpy 的情況下手動排序。有人可以給我一個有效的方法來做這種事情嗎？謝謝
import random
n = 1000
index = list(range(n))
keys = list(range(n))
random.shuffle(keys)
index.sort(key=lambda x: keys[x])) <= HOW TO CONVERT THIS ?
編輯 ：
import numpy as np
from numba import jit
@jit(nopython=True)
def fourier_fit_extra(data, harmonic, extra=0):
size = len(data)
x = np.arange(0, size, 1)
m = np.ones((x.shape[0], 2))
m[:, 1] = x
scale = np.empty((2,))
for n in range(0, 2):
norm = np.linalg.norm(m[:, n])
scale[n] = norm
m[:, n] /= norm
lsf = (np.linalg.lstsq(m, data, rcond=-1)[0] / scale)[::-1]
lsd = data - lsf[0] * x
size_lsd = len(lsd)
four = np.zeros(size_lsd, dtype=np.complex128)
for i in range(size_lsd):
sum_f = 0
for n in range(size_lsd):
sum_f = lsd[n] * np.exp(-2j * np.pi * i * n * (1 / size_lsd))
four[i] = sum_f
freq = np.empty(size)
mi = (size - 1) // 2 1
freq[:mi] = np.arange(0, mi)
freq[mi:] = np.arange(-(size // 2), 0)
freq *= 1.0 / size
lx = np.arange(0, size extra)
out = np.zeros(lx.shape)
# IT'S USED TO SORT FOURIER REALS
index = [v for _, v in sorted([(np.absolute(four[v]), v) for v in list(range(size))])][::-1]
for i in index[:1 harmonic * 2]:
out = (abs(four[i]) / size) * np.cos(2 * np.pi * freq[i] * lx np.angle(four[i]))
return out lsf[0] * lx
uj5u.com熱心網友回復：
對于這種特定型別的輸入，您可以通過以下方式實作排序：
for value in index[:]:
index[keys[value]] = value
如果鍵不是來自 a 的排列range(n)（如您的問題），則創建臨時元組，呼叫sorted然后再次從元組中提取值：
result = [value for _, value in sorted(
[(keys[value], value) for value in index]
)]
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/407412.html
標籤：