python numpy入门学习

环境安装

pip3 install numpy scipy matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

ndarray

在numpy中 数组的维度称为轴,轴的个数称为秩

创建数组

import numpy as np
a = np.arange(6)
print(a)
a = np.arange(2,7,2)
print(a)

结果

创建二维数组:

import numpy as np
b = np.array([[1,2,3],[4,5,6]])  #创建二维数组
print(b)
print(b.shape)
print(b.T)  #输出二维数组b的转置

索引和切片

简单例子:

import numpy as np
a = np.arange(9)
# print(a[2])
# print(a[2:5])  #输出数组下标2到5

b = np.array([[1,2,3],[4,5,6],[7,8,9]])  #创建二维数组
print(b)
#print(b[1])  #输出第一行 (从0开始算)
print()
print(b[1:,])  #输出第一行以及后面所有元素
print()
print(b[1:,[0,1]])  #输出第一行以及后面所有行的第0,1个元素

结果:

简单计算

代码例子:

import numpy as np
a = np.array([1,2,3])
b = np.array([4,5,6])
print(np.add(a,b))
print(np.divide(a,b))
print(np.power(a,b))