编写一个快速函数来检查矩阵中的元素

Question

我正在尝试编写一个函数来检查矩阵中的元素并以布尔值返回另一个结果矩阵。

输入：

X: 二维年龄数组列表，如上所述。

输出：

该函数应返回一个二维数组，其中的条目为 0 或 1，如上所述。

该函数的运行速度必须比这个快 15 倍：

def check_elems(X):
    out = [[0]*len(X[0]) for _ in range(len(X))]
    for i in range(len(X)):
        for j in range(len(X[i])):
            check = X[i][j]
            if check>=14 and check%5==4 and check!=19:
                out[i][j] = 1
    return out

这是一个具体的例子：

• 此人至少 14 岁
• 人的年龄以 4 或 9 结尾
• 而且这个人不是 19 岁

例如一个年龄数组

[[22, 13, 31, 13],
 [17, 14, 24, 22]]

会有输出数组：

[[0, 0, 0, 0], 
 [0, 1, 1, 0]]

Answer 1

没有 numpy：

X = [[22, 13, 31, 13],
     [17, 14, 24, 22]]

%timeit check_elems(X)
%timeit [[1 if i%5==4 and i>=14 and i!=19 else 0 for i in l] for l in X]

使用标准笔记本电脑：

每个循环 2.16 µs ± 15 ns（平均值 ± 标准偏差，7 次运行，每次 100000 次循环）

每个循环 911 ns ± 2.23 ns（平均值 ± 标准偏差，7 次运行，每次 1000000 次循环）

编辑

正如@user3386109 所指出的，前面的代码不满足 OP 要求的 15 倍快 约束，我不知何故误读为 1.5 倍。我将我的代码作为基线。