計算NumPy中數組的平均值

Question

給出三個列表：

A = [1, 2, 3]
B = [4, 5, 6]
C = [7, 8, 9]

如何計算以下（列）的平均值？

• 平均值[1，4，7]
• [2，5，8]的平均值
• 平均值[3，6，9]

不是以下（行），

• [1、2、3]的平均值
• [4，5，6]的平均值
• 平均值[7，8，9]

---

import numpy as np
A = np.array([1,2,3])
B = np.array([4,5,6])
C = np.array([7,8,9])

我該怎么辦？

Answer 1

使用numpy.ndarray.transpose ：

>>> np.array([
...     [1,2,3],
...     [4,5,6],
...     [7,8,9]
... ]).transpose().mean(axis=1)
array([ 4.,  5.,  6.])

或使用numpy.transpose ：

>>> np.transpose([
...     [1,2,3],
...     [4,5,6],
...     [7,8,9]
... ]).mean(axis=1)
array([ 4.,  5.,  6.])

UPDATE

正如Dave Hirschfeld所評論的那樣， axis=0均值要好得多：

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

Answer 2

import numpy as np
A = np.array([1,2,3])
B = np.array([4,5,6])
C = np.array([7,8,9])

Z = zip(A,B,C)

for z in Z :
    print np.mean(np.asarray(z))