最佳方式优化配对(Python)

时间:2017-07-27 12:56:21

标签: python pandas numpy optimization scipy

我正在尝试优化两组数据之间的配对(' A'' B')。我创建了一个Pandas DataFrame,其中包含所有可能的组合组合,以及一些定义特征(pre_optimized = pd.DataFrame(possible_pairs):

A  B     C    D
1  a   23.05  2
1  b  100.00  4
1  c   15.12  7
2  a   50.37  5
2  d   72.65  6
2  c   42.00  1

我想输出最佳配对(post_optimized = pd.DataFrame(optimal_pairs),以便:

maximize(post_optimized['C'].sum())
for x in post_optimized['A'].unique():
     10 <= post_optimized[post_optimized['A']==x]['D'].sum()

我希望输出DataFrame看起来像上面那样,但只有那些被认为是最佳的配对&#39;基于约束:

A  B     C    D
1  a   23.05  2
1  b  100.00  4
2  d   72.65  6
2  c   42.00  1

我相当肯定我需要使用scipy.optimize,但我不知道从哪里开始...

任何指针都将不胜感激!提前谢谢!

0 个答案:

没有答案