numpy基本使用

数组的创建

import numpy as np

t1 = np.array([1,2,3])
print(t1,type(t1))
>>>[1 2 3] <class 'numpy.ndarray'>

t2 = np.array(range(10))
print(t2,type(t2))
>>>[0 1 2 3 4 5 6 7 8 9] <class 'numpy.ndarray'>

t3 = np.arange(10)#跟range用法一样,np.arrage(1,10,2)
print(t3,type(t3))
>>>[0 1 2 3 4 5 6 7 8 9] <class 'numpy.ndarray'>

numpy中常见的数据类型:



查看数据类型:

t1 = np.array(range(1,6))
print(t1)
print(t1.dtype)
>>>[1 2 3 4 5]
>>>int32

指定数据类型:

t1 = np.array(range(1,6),dtype=float) #方式二:dtype="int8"   #方式三:dtype="i1"
print(t1)
print(t1.dtype)
>>>[1. 2. 3. 4. 5.]
>>>float64

#布尔类型
t1 = np.array([1,0,1,1],dtype=bool)
print(t1)
print(t1.dtype)
>>>[ True False  True  True]
>>>bool

修改数据类型:

t1 = np.array([1,0,1,1],dtype=bool)
print(t1)
print(t1.dtype)
>>>[ True False  True  True]
>>>bool

t2 = t1.astype("int8")
print(t2)
print(t2.dtype)
>>>[1 0 1 1]
>>>int8

修改浮点型保留小数位数:

t1 = np.array([random.random() for i in range(3)])
print(t1)
print(t1.dtype)
>>>[0.63296597 0.04263672 0.92288572]
>>>float64

t2 = np.round(t1,2)
print(t2)
>>>[0.63 0.04 0.92]

数组的形状

#一维数组
t1 = np.arange(5)
print(t1)
print(t1.shape)
>>>[0 1 2 3 4]
>>>(5,)

t2 = np.array([[1,2,3],[4,5,6]])
print(t2)
print(t2.shape)
>>>[[1 2 3]
 [4 5 6]]
>>>(2, 3)

修改数组形状,reshape()方法不会改变数组本身,会返回一个新的数组。reshape((12,1))或者reshape(12,1)都行,官方文档使用的是元祖

#一维数组
t1 = np.arange(12)
print(t1)
print(t1.shape)
>>>[ 0  1  2  3  4  5  6  7  8  9 10 11]
>>>(12,)

#t1修改成二维数组,12行1列
t2 = t1.reshape(12,1)
print(t2)
>>>
[[ 0]
 [ 1]
 [ 2]
 [ 3]
 [ 4]
 [ 5]
 [ 6]
 [ 7]
 [ 8]
 [ 9]
 [10]
 [11]]

#t1修改成二维数组,1行12列
t3 = t1.reshape(1,12)
print(t3)
>>>[[ 0  1  2  3  4  5  6  7  8  9 10 11]]

#t1修改成二维数组3行4列
t4 = t1.reshape((3,4))
print(t4)
>>>[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

#t1修改成二维数组3行5列报错
t5 = t1.reshape((3,5))
print(t5)
>>>Traceback (most recent call last):
  File "test.py", line 12, in <module>
    t3 = t1.reshape((3,5))
ValueError: cannot reshape array of size 12 into shape (3,5)

#t4修改成一维数组
t6 = t4.reshape((t4.shape[0]*t4.shape[1],))
print(t6)
>>>[ 0  1  2  3  4  5  6  7  8  9 10 11]

#t4修改成一位数组,简便方法
t7 = t4.flatten()
print(t7)
>>>[ 0  1  2  3  4  5  6  7  8  9 10 11]

数组的计算

t1 = np.arange(12).reshape(3,4)
print(t1)
>>>
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

#每个元素都+10
t2 = t1+10
print(t2)
>>>
[[10 11 12 13]
 [14 15 16 17]
 [18 19 20 21]]

#每个元素都除以0,0理解成非常小的数
t3 = t1/0
print(t3)
>>>
[[nan inf inf inf]
 [inf inf inf inf]
 [inf inf inf inf]]

#相同形状的数组相加,即对应位置的元素相加
t4 = np.arange(12,24).reshape(3,4)
print(t4)
>>>
[[12 13 14 15]
 [16 17 18 19]
 [20 21 22 23]]
t5 = t1 + t4
print(t5)
>>>
[[12 14 16 18]
 [20 22 24 26]
 [28 30 32 34]]

nan表示不是一个数字not a number
inf表示无穷大infinity

#形状不完全相同计算,要么列形状相同,要么行形状相同
t1 = np.arange(12).reshape(3,4)
print(t1)
>>>
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
#含有4个元素的一维数组
t2 = np.array([1,2,3,4]) #t2 = np.array([1,2,3,4]).reshape(1,4) 和一维数组效果一样
print(t2)
>>>
[1 2 3 4]

#t1每行和t2的列数量相同,列对应元素 做减运算
print(t1-t2)
>>>
[[-1 -1 -1 -1]
 [ 3  3  3  3]
 [ 7  7  7  7]]

#3行1列
t3 = np.array([1,2,3]).reshape(3,1)
print(t3)
>>>
[[1]
 [2]
 [3]]
#t1每列和t3行数量相同, 行对应元素 做减运算
print(t1-t3)
>>>
[[-1  0  1  2]
 [ 2  3  4  5]
 [ 5  6  7  8]]

#形状完全不同计算报错,行列形状都不同
t4 = np.array([1,2,3,4,5])
print(t1-t4)
>>>
ValueError: operands could not be broadcast together with shapes (3,4) (5,) 
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,937评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,503评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,712评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,668评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,677评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,601评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,975评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,637评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,881评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,621评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,710评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,387评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,971评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,947评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,189评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,805评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,449评论 2 342

推荐阅读更多精彩内容