将函数应用于列后重建numpy数组

Question

说我有一个numpy数组

import numpy as np
>>> a = np.array([[1, 2], 
                  [3, 4]])

我想提取每一列并像这样应用一个函数

>>> a_col_1 = a[:, 0]
array([1, 3])
>>> new_col_1 = tranform(a_col_1)
array([[1, 1], 
       [3, 3]])

>>> a_col_2 = a[:, 1]
array([2, 4])
>>> new_col_2 = tranform(a_col_2)
array([[2, 2], 
       [4, 4]])

然后以某种方式用新的扩展值代替原始的singe值来重构原始数组，如下所示：

array([[1, 1, 2, 2], 
       [3, 3, 4, 4]])

有没有方便的方法来做到这一点？

Answer 1

这实际上是超级容易的！ 使用numpy.concatenate函数进行的一些快速实验发现，使用np.concatenate([new_col_1, new_col_2], axis=1)可以达到所需的结果。