日韩久久久精品,亚洲精品久久久久久久久久久,亚洲欧美一区二区三区国产精品 ,一区二区福利

講解Python3中NumPy數(shù)組尋找特定元素下標(biāo)的兩種方法

系統(tǒng) 5204 0

 引子  

Matlab中有一個(gè)函數(shù)叫做find,可以很方便地尋找數(shù)組內(nèi)特定元素的下標(biāo),即:Find indices and values of nonzero elements。          
  這個(gè)函數(shù)非常有用。比如,我們想計(jì)算圖1中點(diǎn)Q(x0, y0)拋物線的最短距離。一個(gè)可以實(shí)施的方法是:計(jì)算出拋物線上所有點(diǎn)到Q點(diǎn)的距離,找到最小值,用find函數(shù)找到最小值對(duì)應(yīng)的下標(biāo),即M點(diǎn)橫坐標(biāo)和縱坐標(biāo)對(duì)應(yīng)的元素的下標(biāo),M點(diǎn)到Q點(diǎn)的距離就是最短距離。          

C3TDY]L[JQ`2GLUG2P_E7N5.png               

首先給出Matlab使用find函數(shù)實(shí)現(xiàn)的代碼:

a = linspace(-5,5,1000);
b = a .^2;
x0 = 4;
y0 = 4;
dis = sqrt((a - x0).^2 + (b - y0).^2);
mm = find (dis == min(dis));
a0 = a(mm);
b0 = b(mm);
disMin = sqrt((a0 - x0).^2 + (b0 - y0).^2);
plot(a, b);
hold on;
scatter(x0, y0, 'k*');
scatter(a0, b0, 'k*');
xx = [a0, x0];
yy = [b0, y0];
plot(xx, yy);

SVWH]I{Z~A715Y17OVS3A32.png  

 NumPy中的where函數(shù)  

Syntax: np.where(conditions, [x,y])          

具體實(shí)現(xiàn)代碼如下:

import numpy as np
import math
import matplotlib.pyplot as plt

a = np.linspace(-5, 5, 10000)
b = a * a
x0 = 4
y0 =4
num = np.linspace(0, len(a) - 1, len(a))
dis = np.linspace(0, 0, len(a))
for k in num:
  k = int(k)
  dis[k] = dis[k] + math.sqrt((a[k] -x0) **2 + (b[k] - y0) **2)
disMin = min(dis)
disMinIndex = np.where(dis == disMin)
disMin0 = math.sqrt((a[disMinIndex] - x0) **2 + (b[disMinIndex] - y0) **2)
print('The mininum distance:',disMin)
print('The mininum distance:',disMin0)
print(type(dis))
a0 = a[disMinIndex]
b0 = b[disMinIndex]
fig = plt.figure(figsize = (6,6), dpi = 200)
ax1 = plt.subplot(1,1,1)
line11 = ax1.scatter(a,b,s = 1)
line12 = ax1.scatter(x0, y0, s = 100, marker = '*', color = 'darkorange')
line13 = ax1.scatter(a0, b0, s = 100, marker = '*', color = 'darkorange')
line14 = ax1.plot([x0,a0],[y0,b0], color = 'darkorange')
line15 = ax1.text(4.2,4,'Q(x0,y0)')
line16 = ax1.text(0.6,5, 'M(a0,b0)')
line18 = plt.xlim(-5,5)
line17 = plt.ylim(0,25)
plt.savefig('C:/Users/BRIAR/Desktop/index.png')
plt.show()

The mininum distance: 1.943317035          
  The mininum distance: 1.9433170350024023          
  class ‘numpy.ndarray'          

3LV[G9[AEHTGH_~}EW(2NLD.png  

 List中的index函數(shù)  

Syntax: List.index(aimElement)          
  注意:此處需將NumPy數(shù)組轉(zhuǎn)換成List格式的數(shù)據(jù)。          
  具體實(shí)現(xiàn)代碼如下:

import numpy as np
import math
import matplotlib.pyplot as plt

a = np.linspace(-5, 5, 10000)
b = a * a
x0 = 4
y0 =4
num = np.linspace(0, len(a) - 1, len(a))
dis = np.linspace(0, 0, len(a))
for k in num:
  k = int(k)
  dis[k] = dis[k] + math.sqrt((a[k] -x0) **2 + (b[k] - y0) **2)
disMin = min(dis)
disList = dis.tolist()
disMinIndex = disList.index(disMin)
disMin0 = math.sqrt((a[disMinIndex] - x0) **2 + (b[disMinIndex] - y0) **2)
print('The mininum distance:',disMin)
print('The mininum distance:',disMin0)
print(type(disList))
a0 = a[disMinIndex]
b0 = b[disMinIndex]
fig = plt.figure(figsize = (6,6), dpi = 200)
ax1 = plt.subplot(1,1,1)
line11 = ax1.scatter(a,b,s = 1)
line12 = ax1.scatter(x0, y0, s = 100, marker = '*', color = 'darkorange')
line13 = ax1.scatter(a0, b0, s = 100, marker = '*', color = 'darkorange')
line14 = ax1.plot([x0,a0],[y0,b0], color = 'darkorange')
line15 = ax1.text(4.2,4,'Q(x0,y0)')
line16 = ax1.text(0.6,5, 'M(a0,b0)')
line18 = plt.xlim(-5,5)
line17 = plt.ylim(0,25)
plt.savefig('C:/Users/BRIAR/Desktop/index.png')
plt.show()

The mininum distance: 1.943317035          
  The mininum distance: 1.9433170350024023          
  class ‘list'          

(4Y4%9S90{2APE72~[6VB5C.png  

以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。

 


更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主

微信掃碼或搜索:z360901061

微信掃一掃加我為好友

QQ號(hào)聯(lián)系: 360901061

您的支持是博主寫作最大的動(dòng)力,如果您喜歡我的文章,感覺我的文章對(duì)您有幫助,請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點(diǎn)擊下面給點(diǎn)支持吧,站長(zhǎng)非常感激您!手機(jī)微信長(zhǎng)按不能支付解決辦法:請(qǐng)將微信支付二維碼保存到相冊(cè),切換到微信,然后點(diǎn)擊微信右上角掃一掃功能,選擇支付二維碼完成支付。

【本文對(duì)您有幫助就好】

您的支持是博主寫作最大的動(dòng)力,如果您喜歡我的文章,感覺我的文章對(duì)您有幫助,請(qǐng)用微信掃描上面二維碼支持博主2元、5元、10元、自定義金額等您想捐的金額吧,站長(zhǎng)會(huì)非常 感謝您的哦!!!

發(fā)表我的評(píng)論
最新評(píng)論 總共0條評(píng)論
主站蜘蛛池模板: 雅江县| 沂源县| 阳新县| 康马县| 台州市| 凤山市| 资溪县| 冕宁县| 伊金霍洛旗| 清水河县| 广德县| 米易县| 东莞市| 泰兴市| 左贡县| 广水市| 宁城县| 封开县| 辽宁省| 定远县| 绩溪县| 开平市| 安庆市| 兰溪市| 大埔县| 巴青县| 科尔| 湖口县| 岳池县| 稻城县| 合水县| 绥棱县| 金塔县| 迁西县| 平乡县| 沐川县| 扶沟县| 海城市| 清流县| 宝坻区| 军事|