用数组对数组进行加减运算,然后存储在字典中

时间:2018-11-18 16:18:52

标签: python pandas numpy data-science preprocessor

我已经对该数据进行了大量的预处理和数学运算,以得出两个大小相等的3xN numpy数组。 A和B。

A =整数已被分类为预测B的标签。 B =时间序列数据。

我也有C,它只是B [1:]

A和B在各自的时间步长相等,因此我想不出更好的方法来列出B的下一个时间步长,因此我得到A [0:-1],B [0: -1]和C [1:])

我知道A +或-B在C中总是==至少1个整数。

示例:

如果A = [2,3,4]和B = [5,2,1],我知道C的至少1个将是7,8,1,3或这些数字的任何其他组合+或-彼此。

如何做一个简单的函数来执行此操作,并检查它对C而言是否为==,然后将其存储为字典中的键?目标将是在完成评估新数据后创建一个神经网络。

1 个答案:

答案 0 :(得分:0)

仅举一个例子,如果我了解您的需求:

a = [2,3,4]
b = [5,2,1]
c = set(b[1:])

sums = set([ aa + bb for aa in a for bb in b ])
subs1 = [ (aa - bb) for aa in a for bb in b if (aa - bb) > 0]
subs2 = [ bb - aa for bb in b for aa in a if (bb - aa) > 0]
subs = set(subs1 + subs2)

print(sums)
print(subs)
print(c)

print(len(sums & c)) #=> 0 c does not contains any of sums
print(len(subs & c)) #=> 2 c contains two of subs