Numpy ND 数组到列表列表

Question

我有一个形状为 (5, 641, 2) 的 numpy ND 数组，我想将其转换为列表列表。每个数组都是numpy数组类型，需要转成列表。我需要保持身材。目前我正在用一堆地图函数来做这件事，这真的很慢。

是否有更快/更优雅的方法将 ND numpy 数组转换为列表列表？

def fourth_map(k):
    return k.tolist()

def third_map(l):
    return list(list(map(fourth_map, l)))

def second_map(n):
    return list(map(third_map, n))

def first_map(x):
    return list(map(second_map, x))

lists = np.array(list(map(first_map, arr)))