以下のコードにおきまして、分からない部分があります。
# Welcome to Jupyter! # Welcome to Jupyter!import itertools import itertools import numpy as np import pandas as pd scores = pd.DataFrame( { "番号": [{1}, {2}, {3}, {4}, {5}, {6}, {7}, {8}, {1, 3}, {1, 7}, {3, 6}, {2, 8}], "点数": [70, 40, 50, 30, 90, 15, 35, 65, 20, 30, 12, 40], } ) def get_match_scores(*args): max_len = scores["番号"].apply(len).max() # 点数表のパターンの最大の長さ print('args',args) print(type(args)) # パターンを列挙する patterns = [] for r in range(1, max_len + 1): for p in itertools.combinations(args, r): print(p) patterns.append(set(p)) print('patterns',patterns ) # 該当するパターンの点数を抽出する matches = scores[np.isin(scores["番号"], patterns)] return matches matches = get_match_scores(2, 8) print(matches)
jupyternotebookにて、itertools.combinations
直下のprint(p)
を出力してみたのですが、(2,)(8,)(2, 8)
との結果となりました。
こちらなのですが、combnationsは組み合わせを計算する関数であるかと思いますが、腑に落ちない点があります。
まず、最初のforループでargs
には「(2,8)」が入り、r
には「1」が入ります。
この場合、2,8で考えられる1通りの組み合わせは「1」であり、次のループでは「2」が返ってくる形になるのではないでしょうか。
いまいち、こちらの部分につきまして分からない為、ご助言頂けましたら幸いです。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/09/14 10:46