重塑一个numpy数组答案

【问题标题】：Reshaping a numpy array重塑一个numpy数组
【发布时间】：2012-05-31 12:07:21
【问题描述】：

我想在 iPython 中重塑以下 numpy 数组：

array([[1, 2, 3, 4],[5, 6, 7, 8],[9, 10, 11, 12]]) # Array A

到：

array([[1, 5, 9],[2, 6, 10],[3, 7, 11],[4, 8, 12]]) # Array B

主要任务是计算Array A的第一个元素的平均值（意思是1,5,9的平均值），然后是第二个元素，等等

我认为最简单的方法是重塑数组，然后计算它的平均值。

有什么方法可以在不通过for 循环遍历数组的情况下做到这一点？

【问题讨论】：

【解决方案1】：

在mean 上使用axis 关键字；不用reshape:

>>> A = np.array([[1, 2, 3, 4],[5, 6, 7, 8],[9, 10, 11, 12]])
>>> A.mean(axis=0)
array([ 5.,  6.,  7.,  8.])

如果您确实想要数组B，那么您需要转置数组，而不是重塑它：

>>> A.T
array([[ 1,  5,  9],
       [ 2,  6, 10],
       [ 3,  7, 11],
       [ 4,  8, 12]])

但是你需要将axis=1 给mean。

【讨论】：

【解决方案2】：

要进行这种计算，您应该使用 numpy。

假设 a 是你的起始数组：

a.transpose()

会成功的

【讨论】：