求一多維數(shù)組接近的算法。

dahe_1984

RT A=[3, -1, 4] B = numpy.array([1,1,1],                            [1,-1,1],                            [1,1, -1]) A 在數(shù)組B中最接近的是 [1, -1, 1]， 比較條件：正負(fù)的權(quán)重大于加減，這句話說(shuō)明白了吧。 B1 = numpy.array([1,1,1],                            [1,-1,1],                            [1,1, -1],                            [3,1,4])     結(jié)果還是[1,-1,1]。因?yàn)榧訙p沒(méi)有全匹配到。 數(shù)字中沒(méi)有0.

523066680

1. my $A = [3, -1, 4];
   
2. 
   
3. my $B = [
   
4.     [1, 1, 1],
   
5.     [1,-1, 1],
   
6.     [2,-1, 3],
   
7.     [1, 1,-1],
   
8.     [3, 1, 4]
   
9.    ];
   
10. 
    
11. # 估值
    
12. my @esta;
    
13. my @estb;
    
14. my $e_a;
    
15. my $e_b;
    
16. 
    
17. for my $subset ( @$B )
    
18. {
    
19.     $e_a = 0;
    
20.     $e_b = 0;
    
21.     for my $id ( 0 .. $#$A )
    
22.     {
    
23.         $e_a ++ if ( $subset->[$id] / $A->[$id] < 0 );   # 若正負(fù)相反，估值變大
    
24.         $e_b += ($subset->[$id] - $A->[$id])**2;         # 分量距離作為估值
    
25.     }
    
26. 
    
27.     push @esta, $e_a;
    
28.     push @estb, sqrt($e_b);
    
29. }
    
30. 
    
31. # 顯示各組估值情況
    
32. for my $id ( 0..$#esta )
    
33. {
    
34.     printf "%4.1f %4.1f %4.1f  ", @{$B->[$id]};
    
35.     printf "esta: %d estb: %.3f\n", $esta[$id], $estb[$id];
    
36. }
    
37. 
    
38. # 分級(jí)排序，正負(fù)估值優(yōu)先，向量距離其次
    
39. my @order=
    
40.     sort
    
41.     {
    
42.         $esta[$a] <=> $esta[$b] ||
    
43.         $estb[$a] <=> $estb[$b]
    
44.     } ( 0 .. $#esta );
    
45. 
    
46. my $best = $order[0];
    
47. 
    
48. # 結(jié)果
    
49. printf "\nBest: %s\n", join(", ", @{ $B->[$best] } );
    

復(fù)制代碼

2, -1, 3 是自己加上去的。

1.0  1.0  1.0  esta: 1 estb: 4.123
1.0 -1.0  1.0  esta: 0 estb: 3.606
2.0 -1.0  3.0  esta: 0 estb: 1.414
1.0  1.0 -1.0  esta: 2 estb: 5.745
3.0  1.0  4.0  esta: 1 estb: 2.000

Best: 2, -1, 3

laputa73

這個(gè)需求還是說(shuō)不清,
[-1,-1,1]和[1,-1,-1]哪個(gè)更接近?  如果沒(méi)有正好匹配的,哪個(gè)更接近?
[1,-1,1]和[2,-2,1]哪個(gè)更接近?   正負(fù)匹配, 加減也一樣,怎么算?

dahe_1984

回復(fù) 2# laputa73

看來(lái)是我沒(méi)說(shuō)清楚。重點(diǎn)考慮正負(fù)，沒(méi)有沒(méi)有正負(fù)，返回沒(méi)有匹配到，匹配到正負(fù)后，算出最接近得

dahe_1984

回復(fù) 2# 523066680

多多感謝

dahe_1984

A = np.array([3,-1, 4])

B = np.array([[1,1,1],[1,-1,1],[1,1,-1]])

# distances array-wise
np.abs(B - A)

# sum of absolute values of distances (smallest is closest)
np.sum(np.abs(B - A), axis=1)

# index of smallest (in this case index 1)
np.argmin(np.sum(np.abs(B - A), axis=1))

# all in one line (take array 1 from B)
result = B[np.argmin(np.sum(np.abs(B - A), axis=1))]