亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

從向量中找到最接近的數字索引

從向量中找到最接近的數字索引

ibeautiful 2022-12-20 16:21:11
如果有參考向量,例如:ref = np.array([1., 3., 5.])和一個隨機向量:r = np.array([3.1, 4.7, 0.1, 5.5])對于'r'中的每個數字,有沒有一種快速的方法可以找到'ref'中關閉數字的索引?預期結果是:[2, 3, 1, 3]3.1 最接近 3,所以答案是索引 2。4.7 最接近 5,所以答案是指數 3。0.1 最接近 1,所以答案是索引 1。5.5 最接近 5,所以答案是指數 3。
查看完整描述

1 回答

?
開心每一天1111

TA貢獻1836條經驗 獲得超13個贊

你可以做一個廣播:

np.abs(r[:,None]-ref).argmin(-1)

輸出(記住 python 是從 0 開始索引的):

array([1, 2, 0, 2])


查看完整回答
反對 回復 2022-12-20
  • 1 回答
  • 0 關注
  • 85 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號