Question

我有一个生成渐变的简单函数。我想改变渐变的方向，以对角线左上角。

im = Image.new('RGB', (300, 300))
ld = im.load()

# A map of rgb points in your distribution
# [distance, (r, g, b)]
# distance is percentage from left edge
heatmap = [

    [0.0, (0, 0, 0)],
    [1.00, (0.8, 0.0, 1.0)],
]


def gaussian(x, a, b, c, d=0):
    return a * math.exp(-(x - b) ** 2 / (2 * c ** 2)) + d


def pixel(x, width=100, map=[], spread=1):
    width = float(width)
    r = sum([gaussian(x, p[1][0], p[0] * width, width / (spread * len(map))) for p in map])
    g = sum([gaussian(x, p[1][1], p[0] * width, width / (spread * len(map))) for p in map])
    b = sum([gaussian(x, p[1][2], p[0] * width, width / (spread * len(map))) for p in map])
    return min(1.0, r), min(1.0, g), min(1.0, b)


for x in range(im.size[0]):
    r, g, b = pixel(x, width=300, map=heatmap)
    r, g, b = [int(256 * v) for v in (r, g, b)]
    for y in range(im.size[1]):
        ld[x, y] = r, g, b

我发现我可以从上到下，从左到右改变方向，但不知道如何对角地改变它。

for y in range(im.size[1]):
    ld[y, x] = r, g, b

...给出

for y in range(im.size[1]):
    ld[x, y] = r, g, b

...给出

这是否可以使用当前功能以及如何实现？

Answer 1

您的width应为300 * 300 然后用下面的代码替换你的循环应该可以工作

for y in range(im.size[1]-1):
    r, g, b = pixel(y, width=300*300, map=heatmap)
    r, g, b = [int(256 * v) for v in (r, g, b)]
    for x_ in range(y+1):
        ld[x_,y - x_] = r, g, b

for x in range(im.size[0]):
    r, g, b = pixel(300+x, width=300*300, map=heatmap)
    r, g, b = [int(256 * v) for v in (r, g, b)]
    for y_ in range(im.size[1]):
        ld[x - y_,y_] = r, g, b

没有经过测试，但这样的事情应该有效

Answer 2

尝试这样的事情：

heatmap = [

    [0.0, (0, 0, 0)],
    [1.00, (0.8, 0.0, 1.0)],
]

# x y coordinates of starting point for gradient
start_x = 0
start_y = 0

for x in range(im.size[0]):
    for y in range(im.size[1]):
        # taxicab distance for linear gradient 
        dist = math.fabs(start_x - x) + math.fabs(start_y - y)
        # for circular gradient
        # dist = math.sqrt(math.pow(start_x - x,2) + math.pow(start_y - y,2))
        start_rgb = heatmap[0][1]
        end_rgb = heatmap[1][1]
        dist = dist / (im.size[0] + im.size[1])
        # for circular gradient
        # dist = dist / (math.sqrt(math.pow(im.size[0],2) + math.pow(im.size[1],2)) 
        r, g, b = map(lambda x,y: x+y, map(lambda a: a*dist, start_rgb), map(lambda b: b*dist, end_rgb))
        r, g, b = [int(256 * v) for v in (r, g, b)]
        ld[x, y] = r, g, b

我假设start_x和start_y位于图片的四个角之一。如果不是，请相应地调整start_x - x和start_y - y。

Python改变渐变以对角开始

2 个答案: