假设我有一个函数,它接受一个形状为(m, k)的numpy数组,并且我想将该函数应用于形状为(n, m, k)的numpy数组的每个元素。简单的方法是遍历给定的numpy数组,并将转换后的元素附加到一个形状为(0, m, k)的空numpy数组中
result = np.empty(shape=(0, m, k))
for element in elements:
result = np.append(result, [some_operation(element)], axis=0)在形状为(n, m, k)的numpy数组上应用一些,比如说,“操作”的有效方法是什么?我猜还有一种更“麻木”的方法。
非常感谢。
发布于 2021-04-07 17:19:26
我认为map是最合适的函数
https://stackoverflow.com/questions/66975410
复制相似问题