python - 浮点值的子集总和不起作用

Question

我编写了计算子集总和的代码，它对于整数值来说就像一个魅力，但对于浮点值它没有给出结果。

代码:

import numpy as np
array = np.array([15,1,4,6,3,10,4.2])
num = 8.2

def subsetsum(array,num):

    if num == 0 or num < 1:
        return None
    elif len(array) == 0:
        return None
    else:
        if array[0] == num:
            return [array[0]]
        else:
            with_v = subsetsum(array[1:],(num - array[0])) 
            if with_v:
                return [array[0]] + with_v
            else:
                return subsetsum(array[1:],num)

print('\nList of Values : ',array)
print('\nSum Desired : ',num)
print('\nValues that add up to sum : ',subsetsum(array,num))

整数值的输出

List of Values :  [15  1  4  6  3 10  4]

Sum Desired :  8

Values that add up to sum :  [1, 4, 3]

浮点值的输出

List of Values :  [ 15.    1.    4.    6.    3.   10.    4.2]

Sum Desired :  8.2

Values that add up to sum :  None

如何使用浮点值？

Answer 1

那么你可以简单地使用 np.isclose 考虑 float 点数比较。因此，要解决您的问题，请将相等比较替换为: array[0] == num与:np.isclose(array[0], num) .

同样，您可能希望在开始时进行类似的编辑来修复:num == 0 。