主页 / coding / 问题 / 79331776
Accepted
ChairmanMeow
Asked: 2025-01-06 08:00:20 +0800 CST

使用一个数据框在固定集合中查找匹配组合

  • 772
一个 乙 碳 德 埃
键 1 1 -1
键 2 1 -1
键 3 1 -1
键 4 -1 1
键 5 1 -1
键 6 1 -1
键 7 1 -1
键 8 1 -2 1

最终结果

一个 乙 碳 德 埃
1 -1

假设我们有上述数据框，其中每个键都是用于创建组合以获得所需最终结果的选项。假设您还可以指定可用于实现以下最终结果的最大组合数，那么如何遍历数据框，当一组组合等于最终结果时，它会打印组成组合的所有键以及它所采用的组合数？

例如，假设最大组合数为 3 键组合。那么以下组合将满足最终结果，并且保持低于或等于允许实现该结果的按键组合数

钥匙 2（本身），连击 1

键 4 + 键 5，组合 2

键 3 + 键 8，组合 2

pandas

1 个回答

  1. Best Answer

    假设：

    df = pd.DataFrame({
    'A': [1, None, None, -1, 1, None, None, None],
    'B': [-1, 1, None, 1, None, 1, None, 1],
    'C': [None, -1, 1, None, -1, None, 1, -2],
    'D': [None, None, -1, None, None, -1, None, 1],
    'E': [None, None, None, None, None, None, -1, None]
}, index=['Key 1', 'Key 2', 'Key 3', 'Key 4', 'Key 5', 'Key 6', 'Key 7', 'Key 8'])

final = pd.Series([None, 1, -1, None, None], index=['A', 'B', 'C', 'D', 'E'])

    您可以使用itertools来产生组合，并使用reindex+sum与预期输出进行比较：

    # helper function to produce the combinations
from itertools import combinations, chain
def powerset(iterable, max_set=None):
    s = list(iterable)
    if max_set is None:
        max_set = len(s)
    return chain.from_iterable(combinations(s, r) for r in range(1, max_set+1))

# loop over the combinations 
# reindex and sum
# compare to final with nulls as 0
MAX_N = 3
for c in powerset(df.index, MAX_N):
    if df.reindex(c).sum().eq(final, fill_value=0).all():
        print(c)

    输出：

    ('Key 2',)
('Key 3', 'Key 8')
('Key 4', 'Key 5')
('Key 1', 'Key 2', 'Key 4')

    请注意，这将产生 Key1/Key2/Key4 作为有效组合，因为 Key1/Key4 自我取消并且只有 Key2 有效。

    为了避免这种情况，您可以跟踪产生的组合，并且只保留那些不是已经看到的有效组合的超集的组合：

    MAX_N = 3
seen = set()
for c in powerset(df.index, MAX_N):
    if df.reindex(c).sum().eq(final, fill_value=0).all():
        f = frozenset(c)
        if not any(f > s for s in seen):
            seen.add(f)
            print(c)

    输出：

    ('Key 2',)
('Key 3', 'Key 8')
('Key 4', 'Key 5')
    • 1

相关问题