记录Python学习中numpy模块里的几个函数:arange()、reshape()与argmax()
arange():创建一维数组
In[2]: import numpy as np
In[3]: a=np.arange(6)
In[4]: a
Out[4]:
array([0, 1, 2, 3, 4, 5])
reshape:改变数组维度
In[5]: a.reshape(2,3)
Out[5]:
#改成2行3列
array([[0, 1, 2],
[3, 4, 5]])
In[6]: a.reshape(3,2)
Out[6]:
#改成3行2列
array([[0, 1],
[2, 3],
[4, 5]])
In[7]: a.reshape(-1,6)
Out[7]:
#不知道几行的情况下确定是6列
array([[0, 1, 2, 3, 4, 5]])
In[8]: a.reshape(6,-1)
Out[8]:
#确定是6行,不知道