自用python基础2

二、数组的存储和处理——NumPy模块

2.1 创建数组

多维数组

array(object,dtype=None,copy=True,order=None,subok=False,ndmin=0)

import numpy as np
a = np.array([1,2,3,4])
b = np.array([[1,2,3], [4,5,6],[7,8,9]])
print(a)
print(b)

运行结果：
[1 2 3 4]
[[1 2 3]
 [4 5 6]
 [7 8 9]]

等差数组

arange(start,stop,step,dtype=None)

import numpy as np
d = np.arange(1,20,4)
e = np.arange(1,5)
print(d)    #[1,5,9,13,17]
print(e)    #[0,1,2,3,4]

随机数组

rand()函数创构建的数组中每一个元素都是[0,1)区间内的随机数。

import numpy as np
f = np.random.rand(3)
g = np.random.rand(2,3) 
print(f)    
print(g)

运行结果：
[0.91281291 0.69903541 0.72866193]
[[0.62584457 0.966894   0.85643789]
 [0.18110455 0.83990563 0.20623303]]

randn()函数创建的数组中的元素时符合标准正态分布（均值为0，标准差为1）的随机数

import numpy as np
f = np.random.randn(3) 
print(f)  

randint()函数创建的数组中的元素是指定范围内的随机整数

import numpy as np
f = np.random.randint(1,5,10) #[1,5)区间
e = np.random.randint(1,5,(4,2)) 
print(f)  
print(e)

运行结果：
[2 2 2 3 3 2 4 1 1 3]
[[4 1]
 [1 4]
 [2 4]
 [3 1]]

2.2查看数组属性

import numpy as np
e = np.array([[4 1],[1 4],[3 1]])  
print(arr.e)    #(3,2)
print(arr.e[0]) #查看数组arr的行数：3
print(arr.e[1]) #查看数组arr的列数：2
print(arr.size) #查看数组arr的大小/元素个数：6
print(arr.dtype)#查看数组元素的数据类型：int32
arr1 = arr.astype(float) #数据类型转换
print(arr.ndim) #查看数组arr的维度：2

2.3选取数组元素

一维数组

import numpy as np
arr = np.array(