根据索引分配值的最快方法

Question

在没有循环的情况下执行此类操作的最快方法是什么？

这段代码创建了一条线，每边都有单独的值。

def get_image(img_size = (500,500),angle=90) :
    angle = angle*np.pi/180
    img = np.zeros(img_size)
    center = np.array(img_size)/2

    for x in range(img_size[0]):
        for y in range(img_size[1]):
            cord = np.array((x,y))
            true_cord = cord - center
            x1,y1 = true_cord
            if y1 > np.tan(angle)*x1 :
                img[x,y] = 1

    return img

Answer 1

您可以使用np.mgrid

import numpy as np
import matplotlib.pyplot as plt

def get_image(img_size=(500,500), angle=90):
    r, c = np.array(img_size)/2
    x1, y1 = np.mgrid[-r:r, -c:c]
    return y1 > np.tan(np.radians(angle)) * x1


plt.imshow(get_image((123,123), 23))
plt.axis('off')
plt.show()

输出

根据索引分配值的最快方法

Fastest way to assign values based on their index

python

numpy

matrix