numpy 求TOP k 找出陣列都相同的數

2021-10-19 02:58:52 字數 1233 閱讀 3863

numpy使用的方法

快速從一組資料中找到最大的值

掌握在矩陣中索引的使用方法

import numpy as np

z = np.

arange

(10000

)b = np.random.

shuffle

(z)n=

5print

(z[np.

argpartition

(-z,n)

[:n]

])

z = np.random.

randint(0

,3,(

10,3)

)print

(z)print

(z[:,1

:])print

(z[:,:

-1])

e = np.

all(z[:,

1:]==z[:,

:-1]

,axis=1)

print

(e)

[

9997

9999

9998

9996

9995

]

[[0

22][

210]

[000

][22

1][0

10][

211]

[112

][12

2][1

00][

012]

][[2

2][1

0][0

0][2

1][1

0][1

1][1

2][2

2][0

0][1

2]][

[02]

[21]

[00]

[22]

[01]

[21]

[11]

[12]

[10]

[01]

][false false true false false false false false false false]

**逐句輸出掌握好索引的準確使用

all 和 any 分別對應邏輯關係裡的與或關係

axis= 的使用限定了行運算還是列運算

在乙個陣列中找出top K的數

這個基本都是老生常談了,看資料量有多大,建乙個小根堆,或者用快速排序的方法來完成這個題。下面給出快速排序的方法,直接上 了。include include include void quicksort int arr,int low,int high arr i val quicksort arr,...

找出陣列中相同的部分和不同的部分

var arr1 0,1,2,3,4,5 var arr2 0,4,6,1,3,9 function getarrequal arr1,arr2 return newarr console.log getarrequal arr1,arr2 輸出 4 0,4,1,3 array filter 方法 ...

兩個已排序陣列,找出相同的部分

存在的兩個陣列,已經排好順序,求其相同的部分,有以下幾種求法 1 窮舉法 最原始的方法,時間複雜度為o m n 如下 int a new int 示列陣列 int b new int for int i 0 i a.length i 2 binary search 通過乙個陣列的for迴圈,不斷與另...