辞書キーのサブセット

Question

私は{'ip1:port1' : <value>, 'ip1:port2' : <value>, 'ip2:port1' : <value>, ...}という形のpython辞書を持っています。ディクショナリキーは、IP：ポートのペアで構成される文字列です。値はこのタスクにとって重要ではありません。

私は一意のIPアドレスを持つip:portの組み合わせのリストが必要です。ポートは元のキーに表示されるポートのいずれかにすることができます。上記の例では、['ip1:port1', ip2:port1']と['ip1:port2', ip2:port1']の2種類が許容されます。

これを実行するにあたって最も厄介な方法は何ですか？

現在、私の解決策は、それが追加のリストを作成し、それらを捨てるので、私は、それを好きではない

def get_uniq_worker_ips(workers): 
    wip = set(w.split(':')[0] for w in workers.iterkeys()) 
    return [[worker for worker in workers.iterkeys() if worker.startswith(w)][0] for w in wip] 

です。

Answer 1

あなたは、同じIPアドレスでグループにitertools.groupbyを使用することができます。

data = {'ip1:port1' : "value1", 'ip1:port2' : "value2", 'ip2:port1' : "value3", 'ip2:port2': "value4"} 
by_ip = {k: list(g) for k, g in itertools.groupby(sorted(data), key=lambda s: s.split(":")[0])} 
by_ip 
# {'ip1': ['ip1:port1', 'ip1:port2'], 'ip2': ['ip2:port1', 'ip2:port2']} 

それからちょうどIPアドレスの異なるグループからいずれかを選択します。グループからちょうど最初のキーのジェネレータ式作っ

{v[0]: data[v[0]] for v in by_ip.values()} 
# {'ip1:port1': 'value1', 'ip2:port1': 'value3'} 

も短く、：しかし

one_by_ip = (next(g) for k, g in itertools.groupby(sorted(data), key=lambda s: s.split(":")[0])) 
{key: data[key] for key in one_by_ip} 
# {'ip1:port1': 'value1', 'ip2:port1': 'value3'} 

は、groupbyをソートする入力データを必要とすることに注意してください。だから、あなたが辞書の中のすべてのキーを並べ替えるのを避けたいのであれば、すでに見ているキーのsetを使うだけです。

seen = set() 
not_seen = lambda x: not(x in seen or seen.add(x)) 
{key: data[key] for key in data if not_seen(key.split(":")[0])} 
# {'ip1:port1': 'value1', 'ip2:port1': 'value3'} 

これはあなたのソリューションに似ていますが、代わりにユニークなキーをループし、それぞれの辞書で一致するキーを見つける、あなたループキーとすでにIPを見てきたかどうかを確認してください。

Answer 2

私のソリューションでは文字数が変更されていて、満足しています。

def get_uniq_worker_ips(workers): 
    wip = set(w.split(':')[0] for w in workers.iterkeys()) 
    return [next(worker for worker in workers.iterkeys() if worker.startswith(w)) for w in wip] 

@Ignacio Vazquez-Abramsおよび@M.T。説明のために。

Answer 3

これを行う1つの方法は、等価性テストを行うときに文字列のIP部分だけを見るカスタムクラスにキーを変換することです。また、適切な__hash__メソッドを提供する必要があります。

ここでのロジックは、比較のポート部分を無視して、同じIPを持つキーを「参照」するため、そのIPを持つキーがすでに存在する場合にキーをセットに追加しないようにすることですセット内に存在する。ここで

は、後でのPython 2.7以上を実行している場合、機能が設定された理解を代わりに使用することができます

class IPKey(object): 
    def __init__(self, s): 
     self.key = s 
     self.ip, self.port = s.split(':', 1) 

    def __eq__(self, other): 
     return self.ip == other.ip 

    def __hash__(self): 
     return hash(self.ip) 

    def __repr__(self): 
     return 'IPKey({}:{})'.format(self.ip, self.port) 

def get_uniq_worker_ips(workers): 
    return [k.key for k in set(IPKey(k) for k in workers)] 

# Test 

workers = { 
    'ip1:port1' : "val", 
    'ip1:port2' : "val", 
    'ip2:port1' : "val", 
    'ip2:port2' : "val", 
} 

print(get_uniq_worker_ips(workers))  

出力

['ip2:port1', 'ip1:port1'] 

のPython 2やPython 3上で動作するいくつかのコードですset()コンストラクタ呼び出し内のそのジェネレータ式の値。

def get_uniq_worker_ips(workers): 
    return [k.key for k in {IPKey(k) for k in workers}] 

IPKey.__repr__方法は厳密には必要ではないですが、私はそれは開発中に便利なことができるので、すべての私のクラス__repr__を与えたいです。ここで

---

は、Jon Clementsの礼儀は非常に効率的であるはるかに簡潔なソリューションです。それは、辞書の理解を介して所望のリストを構築する。

def get_uniq_worker_ips(workers): 
    return list({k.partition(':')[0]:k for k in workers}.values())