Ánh xạ các giá trị trong từ điển python

Đưa ra một từ điển { k1: v1, k2: v2 ... }tôi muốn được { k1: f(v1), k2: f(v2) ... }cung cấp tôi vượt qua một chức năng f.

Có bất kỳ chức năng được xây dựng như vậy? Hay tôi phải làm

dict([(k, f(v)) for (k, v) in my_dictionary.iteritems()])

Lý tưởng nhất là tôi sẽ viết

my_dictionary.map_values(f)

hoặc là

my_dictionary.mutate_values_with(f)

Đó là, nó không quan trọng với tôi nếu từ điển gốc bị đột biến hoặc một bản sao được tạo.

Không có chức năng như vậy; cách dễ nhất để làm điều này là sử dụng một cách hiểu chính tả:

my_dictionary = {k: f(v) for k, v in my_dictionary.items()}

Trong python 2.7, sử dụng .iteritems()phương thức thay vì .items()để tiết kiệm bộ nhớ. Cú pháp hiểu chính tả không được giới thiệu cho đến khi python 2.7.

Lưu ý rằng không có phương pháp như vậy trong danh sách; bạn phải sử dụng map()chức năng hiểu danh sách hoặc chức năng.

Như vậy, bạn cũng có thể sử dụng map()chức năng để xử lý lệnh của mình:

my_dictionary = dict(map(lambda kv: (kv[0], f(kv[1])), my_dictionary.iteritems()))

nhưng đó không phải là có thể đọc được, thực sự.

Những toolz này là tuyệt vời cho loại logic đơn giản nhưng lặp đi lặp lại.

http://toolz.readthedocs.org/en/latest/api.html#toolz.dicttoolz.valmap

Đưa bạn đến đúng nơi bạn muốn.

import toolz
def f(x):
  return x+1

toolz.valmap(f, my_list)

Bạn có thể thực hiện việc này tại chỗ, thay vì tạo một từ điển mới, có thể thích hợp hơn cho các từ điển lớn (nếu bạn không cần một bản sao).

def mutate_dict(f,d):
    for k, v in d.iteritems():
        d[k] = f(v)

my_dictionary = {'a':1, 'b':2}
mutate_dict(lambda x: x+1, my_dictionary)

kết quả trong việc my_dictionarychứa:

{'a': 2, 'b': 3}

Do PEP-0469 đã đổi tên iteritems () thành các mặt hàng () và PEP-3113 đã loại bỏ tham số Tuple giải nén , trong Python 3.x, bạn nên viết Martijn Pieters ♦ trả lời như sau:

my_dictionary = dict(map(lambda item: (item[0], f(item[1])), my_dictionary.items()))

Trong khi câu trả lời ban đầu của tôi bị mất điểm (bằng cách cố gắng giải quyết vấn đề này bằng giải pháp cho khóa Truy cập trong nhà máy của defaultdict ), tôi đã làm lại nó để đề xuất một giải pháp thực tế cho câu hỏi hiện tại.

Đây là:

class walkableDict(dict):
  def walk(self, callback):
    try:
      for key in self:
        self[key] = callback(self[key])
    except TypeError:
      return False
    return True

Sử dụng:

>>> d = walkableDict({ k1: v1, k2: v2 ... })
>>> d.walk(f)

Ý tưởng là phân lớp dict gốc để cung cấp cho nó chức năng mong muốn: "ánh xạ" một hàm trên tất cả các giá trị.

Điểm cộng là từ điển này có thể được sử dụng để lưu trữ dữ liệu gốc như thể nó là một dict, trong khi chuyển đổi bất kỳ dữ liệu nào theo yêu cầu với một cuộc gọi lại.

Tất nhiên, hãy thoải mái đặt tên cho lớp và hàm theo cách bạn muốn (tên được chọn trong câu trả lời này được lấy cảm hứng từ array_walk()chức năng của PHP ).

Lưu ý: Cả khối try- exceptvà returncâu lệnh đều không bắt buộc đối với chức năng, chúng ở đó để bắt chước thêm hành vi của PHP array_walk.

Để tránh thực hiện lập chỉ mục từ bên trong lambda, như:

rval = dict(map(lambda kv : (kv[0], ' '.join(kv[1])), rval.iteritems()))

Bạn cũng có thể làm:

rval = dict(map(lambda(k,v) : (k, ' '.join(v)), rval.iteritems()))

Chỉ cần đi qua trường hợp sử dụng này. Tôi đã triển khai câu trả lời của gens , thêm một cách tiếp cận đệ quy để xử lý các giá trị cũng là các giá trị:

def mutate_dict_in_place(f, d):
    for k, v in d.iteritems():
        if isinstance(v, dict):
            mutate_dict_in_place(f, v)
        else:
            d[k] = f(v)

# Exemple handy usage
def utf8_everywhere(d):
    mutate_dict_in_place((
        lambda value:
            value.decode('utf-8')
            if isinstance(value, bytes)
            else value
        ),
        d
    )

my_dict = {'a': b'byte1', 'b': {'c': b'byte2', 'd': b'byte3'}}
utf8_everywhere(my_dict)
print(my_dict)

Điều này có thể hữu ích khi xử lý các tệp json hoặc yaml mã hóa chuỗi dưới dạng byte trong Python 2