python - 将 Numpy 数组缩放到一定范围

Question

类似于this question ，我想将 Numpy 数组放入某个范围内，但是与链接的问题不同，我不想对其进行规范化。我怎样才能有效地做到这一点？ Numpy 中是否有内置方法？

用一个例子来说明，my_scale 是我正在寻找的函数，out_range 定义了输出范围:

res = my_scale(np.array([-3, -2, -1], dtype=np.float), out_range)
assert res == [-1, 0, 1]
assert res != [-1, -2/3, -1/3]

Answer 1

在 asking on CodeReview 之后, 我被告知有一个内置的 np.interp完成这个:

np.interp(a, (a.min(), a.max()), (-1, +1))

为了后代，我在下面留下了我的旧答案。

---

我根据 this answer 中的 D3.js 代码制作了自己的函数:

import numpy as np

def d3_scale(dat, out_range=(-1, 1)):
    domain = [np.min(dat, axis=0), np.max(dat, axis=0)]

    def interp(x):
        return out_range[0] * (1.0 - x) + out_range[1] * x

    def uninterp(x):
        b = 0
        if (domain[1] - domain[0]) != 0:
            b = domain[1] - domain[0]
        else:
            b =  1.0 / domain[1]
        return (x - domain[0]) / b

    return interp(uninterp(dat))

print(d3_scale(np.array([-2, 0, 2], dtype=np.float)))
print(d3_scale(np.array([-3, -2, -1], dtype=np.float)))