NumPy 中的axis

axis 是多维数组的轴，指定某个轴做运算。

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import numpy as np
inX = np.array([[1,2,1],[0,3,1],[2,1,4],[1,3,1]])
print np.sum(inX, axis=1)
#axis=1时是第二维，即一行一行为单位
print np.min(inX, axis=0)
#axis=0时是第一维，即一列一列为单位