数组基本操作
本节介绍加法、减法、乘法、除法等
一旦创建了数组,就可以开始使用它们了。
例如,假设您创建了两个数组,一个称为“data”,另一个称为“one”
>>> data = np.array([1, 2])
>>> ones = np.ones(2, dtype=int)
>>> data + ones
array([2, 3])
>>> data - ones
array([0, 1])
>>> data * data
array([1, 4])
>>> data / data
array([1., 1.])
NumPy的基本操作很简单。如果要查找数组中元素的和,可以使用sum()
。这适用于一维阵列、二维阵列和高维阵列。
>>> a = np.array([1, 2, 3, 4])
>>> a.sum()
10
要在二维数组中添加行或列,可以指定轴。
如果从这个数组开始:
>>> b = np.array([[1, 1], [2, 2]])
您可以将行相加:
>>> b.sum(axis=0)
array([3, 3])
您可以将列相加为:
>>> b.sum(axis=1)
array([2, 4])
NumPy入门系列教程:
如何将一维array转换为二维array(如何向数组添加新轴)