numpy ndarray掩碼操作

2021-10-04 22:53:32 字數 1181 閱讀 9408

bool掩碼

掩出位置為true處的值

從大資料集中抽取出一小部分(e.g.抽取年齡大於40歲的學生)

import numpy as np

a = np.arange(1,

10)#設定掩碼

mask =

[true

,false

,true

,false

,true

,false

,true

,false

,true

,false

]#如果總括號中放的列表中都是true/false,他就是乙個bool掩碼

#true的位置會漏出原數字

print

(a[mask]

)#返回true位置的值[1 3 5 7 9]

示例1
import numpy as np

ary = np.arange(1,

10)print

(ary)

#設定掩碼

mask = ary %2==

0print

(mask)

#[false true false true false true false]

print

(ary[mask]

)#[2 4 6 8],掩出來的陣列還是原陣列的一部分

ary[mask]

=100

#對掩出來的部分賦值,是修改原陣列

print

(ary)

#[1 100 3 100 5 100 7 100 9]

示例2:輸出100以內3或7的倍數
a = np.arange(

100)

print

(a[(a %3==

0)|(a %7==

0)])

索引掩碼

找出對應下標的元素輸出出來

找出對應下標的元素輸出出來

a = np.array(67,

43,65,

97,33,

11,99)

mask =[3

,6,2

,1]print

(a[mask]

)#[97,99,65,43]

認識Numpy Ndarray物件

numpy numerical python 是 python 語言的乙個擴充套件程式庫,支援大量的維度陣列與矩陣運算,此外也針對陣列運算提供大量的數學函式庫。numpy為什麼能夠受到各個資料科學從業人員的青睞與追捧,其實很大程度上是因為numpy在向量計算方面做了很多優化,介面也非常友好。而這些其...

Numpy ndarray的矩陣處理

1.向量運算 相同大小的陣列間運算應用在元素上 向量與向量運算 arr np.array 1,2,3 4,5,6 print 元素相乘 print arr arr print 矩陣相加 print arr arr 結果 元素相乘 1 4 9 16 25 36 矩陣相加 2 4 6 8 10 12 向...

Numpy Ndarray 常用物件屬性

用於返回陣列的維數,等於秩 import numpy as np a np.arange 24 print a.ndim a 現只有乙個維度 輸出1 現在調整其大小 b a.reshape 2,4,3 b 現在擁有三個維度 print b.ndim 輸出3表示陣列的維度,返回乙個元組,這個元組的長度...