从多个字典列表中有条件地提取数据的大多数 Python 方法-解网

问：

我正在尝试使用两个字典列表来构建一个对象，这些字典是从与两个不同数据库的TAP连接构建的。由于数据源的原因，我不能保证任何词典都会包含我需要的信息，所以我选择了一本主要词典，如果信息不在那里，那么我就从第二本词典中提取它。

由于我从两个不同的数据源中提取数据，因此两个源的 TAP 中的字段名称不同，因此我不能只对字典进行交集。

目前我可以让它工作，但我对解决方案不满意：

for result in eeuresults:
    name=result['target_name']
    axes=result['semi_major_axis']
    period=result['period']
    radius=result['radius']

    if math.isnan(period):
        for r in eparesults:
            if r['pl_name'] == name: period=r['pl_orbper']
    if math.isnan(axes):
        for r in eparesults:
            if r['pl_name'] == name: axes=r['pl_orbsmax']        
    if math.isnan(radius):
        for r in eparesults:
            if r['pl_name'] == name: radius=r['pl_radj']

我尝试使用 dictionary.get（）来简化它，但如果该值不在第二个字典列表中，它就会下降。

axes=result.get('semi_major_axis',[r['pl_orbper'] for r in eparesults if r['pl_name']==name][0])

Python 字典

def find_in_eparesults(name, value, epa_key):
  if math.isnan(value):
    for r in eparesults:
      if r['pl_name'] == name:
        value = r[epa_key]
        break
  return value

for result in eeuresults:
    name=result['target_name']
    axes=find_in_eparesults(name, result['semi_major_axis'], 'pl_orbper')
    period=find_in_eparesults(name, result['period'], 'pl_orbsmax')
    radius=find_in_eparesults(name, result['radius'], 'pl_radj')

eeuresults = [
    {"target_name": "A", "semi_major_axis": 1, "period": 2},
    {"target_name": "B", "period": 3, "radius": 4},
]


eparesults = [
    {"pl_name": "A", "pl_orbper": 1, "pl_orbsmax": 2, "pl_radj": 3},
    {"pl_name": "B", "pl_orbper": 4, "pl_orbsmax": 5, "pl_radj": 6},
]

# This is the important part, convert `eparesults` to a dict:
eparesults_dict = {d["pl_name"]: d for d in eparesults}

for result in eeuresults:
    name = result["target_name"]
    axes = result.get("semi_major_axis", eparesults_dict.get(name, {}).get("pl_orbper"))
    period = result.get("period", eparesults_dict.get(name, {}).get("pl_orbsmax"))
    radius = result.get("radius", eparesults_dict.get(name, {}).get("pl_radj"))

    print(f"{name=} {axes=} {period=} {radius=}")

指纹：

name='A' axes=1 period=2 radius=3
name='B' axes=4 period=3 radius=4

from collections import defaultdict
from math import nan, isnan

class Result(object):
    def __getattr__(self, name):
        return nan
    def __setattr__(self, name, value):
        if isnan(value):
            return
        super().__setattr__(name, value)
    def __iter__(self):
        return iter((self.axes, self.period, self.radius))
    
results = defaultdict(Result)

for d in eeuresults:
    r = results[d['target_name']]
    r.axes   = d['semi_major_axis']
    r.period = d['period']
    r.radius = d['radius']
    
for d in eparesults:
    r = results[d['pl_name']]
    r.axes   = d['pl_orbsmax']
    r.period = d['pl_orbper']
    r.radius = d['pl_radj']

# In case you would need to lookup keys yourself, conversion to dict is needed.
# results = dict(results)
    
for name, (axes, period, radius) in results.items():
    print(f"{name=} {axes=} {period=} {radius=}")

只要您知道它们的密钥，这可以包含任意数量的列表。

此外，如果您想要除 .__setattr__isnan

如果您预计任何字典中缺少密钥，请将 d['key'] 替换为 d.get（'key'， nan）

从多个字典列表中有条件地提取数据的大多数 Python 方法

Most Pythonic method of conditionally extracting data from multiple lists of dictionaries

评论

评论

评论

评论