前言
NumPy模块的主要特点是引入了数组的概念。没错就是c、c++中的数组,不过功能上要强一点。
数组的创建
1、numpy.array(列表)
import os
import xlwings as xw
import numpy as np
# 创建一维数组
a = np.array([1,2,3,4,5])
# 创建二维数组
b = np.array([[1,2],[3,4],[5,6]])
print(a)
print(b)
2、arange()函数
该函数是用来创建一维数组的。
import os
import xlwings as xw
import numpy as np
#一个参数:起点默认为0,步长取默认值1,左闭右开
x = np.arange(5)
#两个参数:第一个参数为起点,第二个参数为终点,步长取默认值1,左闭右开
y = np.arange(5,10)
#三个参数:第一个参数为起点,第二个参数为终点,第三个参数为步长,左闭右开
z = np.arange(5,10,2)
3、用random中的函数创建随机一维数组
import os
import xlwings as xw
import numpy as np
c = np.random.randn(5)
print(c)
4、创建二维数组
可以利用一维数组的arange()
函数和reshape()
函数来创建。例如,将包含0~11这12个整数的一维数组转换成3行4列的二维数组。演示代码如下:
import numpy as np
d = np.arange(12).reshape(3,4)
print(d)
运行结果
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
5、创建按随机整数二维数组
import numpy as np
e = np.random.randint(0,10,(4,4))
print(e)
第2行代码中的np. random. radiant()函数用来创建随机整数,括号里第1个参数0为起始数,第2个参数10为终止数,第3个参数(4.4)则表示创建一个4行4列的二维数组。运行结果如下:
[[6 7 3 5]
[5 9 7 8]
[8 3 3 3]
[9 8 5 9]]