科学计算工具-numpy

简介

Numpy:提供了一个在Python中做科学计算的基础库,重在数值计算,主要用于多维数组(矩阵)处理的库。
用来存储和处理大型矩阵,比Python自身的嵌套列表结构要高效的多。
本身是由C语言开发,是个很基础的扩展,Python其余的科学计算扩展大部分都是以此为基础。
高性能科学计算和数据分析的基础包
ndarray,多维数组(矩阵),具有矢量运算能力,快速、节省空间
矩阵运算,无需循环,可完成类似Matlab中的矢量运算
线性代数、随机数生成

#导入
import numpy as np

数组创建

NumPy数组是一个多维的数组对象(矩阵),称为ndarray,具有矢量算术运算能力,并具有执行速度快和节省空间的特点。

注意:ndarray的下标从0开始,且数组里的所有元素必须是相同类型

ndarray属性

  • ndim属性:维度个数
  • shape属性:维度大小
  • dtype属性:数据类型

ndarray的随机创建

通过随机抽样(numpy.random)生成随机数据

np.random.rand()
随机样本位于[0,1)中,rand固定取件0.0~1.0

1-1

np.random.randn()
随机样本是从标准正态分布中返回一个或多个样本值

1-2

np.random.randint()
生成指定维度大小(3行4列)的随机多维整型数据(二维),randint()可以指定区间(-1, 5)

1-3

np.random.uniform()
生成指定维度大小(3行4列)的随机多维浮点型数据(二维),uniform()可以指定区间(-1, 5)

1-4

ndarray的序列创建

np.array(collection)
collection为序列型对象(list)、嵌套序列对象(list of list)

#list序列转换为ndarray
list = range(10)
arr = np.array(list)
print(arr)#ndarray数据
print(type(arr))
2-1

np.zeros()指定大小的全0数组。注意第一个参数是数组,用来指定大小,如(3,4)

zeros_arr = np.zeros((3,4))
2-2

np.ones()
指定大小全是1的数组,注意:第一个参数是远足,用来指定大小,如(3,4)

np.ones()
2-3

np.empty()
初始化数组,不是总是返回全0,有时返回的是未初始化的随机值(内存里的随机值)

#np.empty
empty_arr = np.empty((3,3))
#np.empty 指定数据类型
empty_int_arr = np.empty((3,3),int)
2-4

np.arange()和reshape()
arange()类似python的range(),创建一个一维ndarray数组
reshape将重新调整数组的维数

# np.arange() 
arr = np.arange(15) # 15个元素的 一维数组 
print(arr) 
print(arr.reshape(3, 5)) # 3x5个元素的 二维数组
print(arr.reshape(1, 3, 5)) # 1x3x5个元素的 三维数组

2-5

ndarray数据类型

dtype参数
指定数组的数据类型,类型名+位数,如float64,int32

#初始化3行4列数组,数据类型为float64
zeros_float_arr = np.zeros((3,4),dtype=np.float64)
print(zeros_float_arr)
3-1

astype方法

#astype转换数据类型,将已有的数组的数据类型转换为int32
zeros_int_arr = zeros_float_arr.astype(np.int32) 
print(zeros_int_arr) 
print(zeros_int_arr.dtype)


3-2

ndarray的矩阵运算

数组是编程中的概念,矩阵、矢量是数学概念。

在计算机编程中,矩阵可以用数组形式定义,矢量可以用结构定义!

1.矢量运算:相同大小的数组间运算应用在元素上

#矢量与矢量运算
arr = np.array([[1,2,3],[4,5,6]])
print("元素相乘:")
print(arr*arr)
print("矩阵相加:")
print(arr + arr)
4-1

2.矢量和标量运算:"广播"-将标量"广播"到各个元素

#矢量与标量运算
arr = np.array([[1,2,3],[4,5,6]])
print(1./arr)
print(2.*arr)
4-2

ndarray的索引与切片

1.一维数组的索引与切片-与python的列表索引功能相似

#一维数组
arr1 = np.arange(10)
print(arr1)
print(arr1[2:5])
5-1

2.多维数组的索引与切片

arr2 = np.random.randint(1,5,size(3,5))
print(arr2)
print(arr2[2:3])
5-2

3.条件索引
布尔值多维数组:arr[condition],condition也可以是多个条件组合。
注意,多个条件组合要使用 & | 连接,而不是Python的 and or。

# 条件索引 找出 data_arr 中 2005年后的数据 
data_arr = np.random.rand(3,3) 
print(data_arr) 
year_arr = np.array([[2000, 2001, 2000], [2005, 2002, 2009], [2001, 2003, 2010]]) 
is_year_after_2005 = year_arr >= 2005 
print(is_year_after_2005, is_year_after_2005.dtype)

5-3
filtered_arr = data_arr[is_year_after_2005] 
print(filtered_arr) 
# 多个条件 
filtered_arr = data_arr[(year_arr <= 2005) & (year_arr % 2 == 0)] 
print(filtered_arr)

5-4

ndarray的转置

二维数组直接使用转换函数:transpose()
高维数组转换要指定维度编号参数 (0, 1, 2, …),注意参数是元组

arr = np.random.rand(2,3) # 2x3 数组 
print(arr) 
print(arr.transpose()) # 转换为 3x2 数组 
5-5
arr3d = np.random.rand(2,3,4) # 2x3x4 数组,2对应0,3对应1,4对应2 
print(arr3d) 
print(arr3d.transpose((1,0,2))) # 根据维度编号,转为为 3x2x4 数组
5-6

元素计算函数

  • ceil(): 向上最接近的整数,参数是 number 或 array
  • floor(): 向下最接近的整数,参数是 number 或 array
  • rint(): 四舍五入,参数是 number 或 array
  • isnan(): 判断元素是否为 NaN(Not a Number),参数是 number 或 array
  • multiply(): 元素相乘,参数是 number 或 array
  • divide(): 元素相除,参数是 number 或 array
  • abs():元素的绝对值,参数是 number 或 array
  • where(condition, x, y): 三元运算符,x if condition else y
# randn() 返回具有标准正态分布的序列。 
arr = np.random.randn(2,3) 
print(arr) 
print(np.ceil(arr)) 
print(np.floor(arr)) 
print(np.rint(arr)) 

6-1
print(np.isnan(arr)) 
print(np.multiply(arr, arr)) 
print(np.divide(arr, arr)) 
print(np.where(arr > 0, 1, -1))
6-2

元素统计函数

  • np.mean(), np.sum():所有元素的平均值,所有元素的和,参数是 number 或 array
  • np.max(), np.min():所有元素的最大值,所有元素的最小值,参数是 number 或 array
  • np.std(), np.var():所有元素的标准差,所有元素的方差,参数是 number 或 array
  • np.argmax(), np.argmin():最大值的下标索引值,最小值的下标索引值,参数是 number 或 array
  • np.cumsum(), np.cumprod():返回一个一维数组,每个元素都是之前所有元素的 累加和 和 累乘积,参数是 number 或 array
  • 多维数组默认统计全部维度,axis参数可以按指定轴心统计,值为0则按列统计,值为1则按行统计。
arr = np.arange(12).reshape(3,4) 
print(arr) 
print(np.sum(arr)) # 所有元素的和 
print(np.sum(arr, axis=0)) # 数组的按列统计和 
print(np.sum(arr, axis=1)) # 数组的按行统计和
print(np.cumsum(arr)) # 返回一个一维数组,每个元素都是之前所有元素的 累加和 
7-1
arr = np.random.randint(0,10,(3,4))
print(arr)
#所有元素的平均值与和
print(np.mean(arr))
print(np.sum(arr))
#求数组的最大值
print(np.max(arr))
print(np.min(arr))
#方差和标准差:衡量数据和期望值之间的偏离值
#求方差:所有元素都和平均数的差的平方的平均数
print(np.var(arr))
#求标准差:方差的平方根
print(np.std(arr))

7-2
#求最大值和最小值的索引下标
#如果是多维数组,将多维数组合并成一维数组,再找出最大值索引下标
arr = np.arange(10)
print(arr)
print(np.argmax(arr))
print(np.argmin(arr))
#返回一维数组,数组每个元素都是之前每个元素的累加和
print(np.cumsum(arr))
#返回一个一维数组,数组的每一个元素都是之前所有元素的累加积
print(np.cumprod(arr))
7-3
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,948评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,371评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,490评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,521评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,627评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,842评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,997评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,741评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,203评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,534评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,673评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,339评论 4 330
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,955评论 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,770评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,000评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,394评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,562评论 2 349

推荐阅读更多精彩内容