获取 yaml 文件中的字典名称

Question

我有一个如下所示的 yaml 文件：

Histograms:
  zee: 
    bins: {"fix": [55, 0.1 ,35e3]}  
    Title: "é isso ai"
    Xlabel: "como assim" 
    Ylabel: "Z #rightarrow e^{+}e^{-}"
    ymax: 35e3
    ymim: 0.1  
    color: 4 #root.kAzure+3 
    scale: 1.0

  ztautau: 
    ymax: 35e3
    ymim: 0.1    
    ymaxp: 1.19
    ymimp: 0.81
    label: "Z #rightarrow e^{+}e^{-}"     
    color: ""

  ttbar: 
    ymax: 35e3
    ymim: 0.1    
    ymaxp: 1.19
    ymimp: 0.81
    label: "Z #rightarrow e^{+}e^{-}"    
    
  wplustaunu: 
     bins: {"fix": [55, 0.1 ,35e3]}  
     Title: "é isso ai"
     Xlabel: "como assim" 
     Ylabel: "Z #rightarrow e^{+}e^{-}"
     ymax: 35e3
     ymim: 0.1  
     color: "root.kAzure+3" 
     scale: 1.0   

我需要通过循环获取以下字符串：zee,ztautau,ttbar,wplustaunu（我猜它们是字典的标题？）。我的意思是我需要通过一个循环（在 python 中）来处理它们，因为我需要处理有很多这样的字典的文件，所以我自己知道它们不是一个好的选择。但是我一直在研究这个问题并且无法找到解决方案，所以如果有人可以帮助我或给我一些参考，我将非常感激。

我想要的输出是：

泽
兹陶陶
ttbar
wplusenu

我不想要更多，只是这些数据。有可能吗？

感谢关注！

Answer 1

您只需要在直方图下过滤掉什么是字典。您可以使用dictionary comprehension。

from yaml import safe_load as yload

di = yload("""    Histograms:
      zee: 
        bins: {"fix": [55, 0.1 ,35e3]}  
        Title: "é isso ai"
        Xlabel: "como assim" 
        Ylabel: "Z #rightarrow e^{+}e^{-}"
        ymax: 35e3
        ymim: 0.1  
        color: 4 #root.kAzure+3 
        scale: 1.0
    
      ztautau: 
        ymax: 35e3
        ymim: 0.1    
        ymaxp: 1.19
        ymimp: 0.81
        label: "Z #rightarrow e^{+}e^{-}"     
        color: ""
    
      ttbar: 
        ymax: 35e3
        ymim: 0.1    
        ymaxp: 1.19
        ymimp: 0.81
        label: "Z #rightarrow e^{+}e^{-}"    
        
      wplustaunu: 
         bins: {"fix": [55, 0.1 ,35e3]}  
         Title: "é isso ai"
         Xlabel: "como assim" 
         Ylabel: "Z #rightarrow e^{+}e^{-}"
         ymax: 35e3
         ymim: 0.1  
         color: "root.kAzure+3" 
         scale: 1.0   
""")

dict_of_dict = {k:v for k,v in di.get("Histograms",{}).items() if isinstance(v, dict)}

for k,v in dict_of_dict.items():  
    print(f"\n{k}:{v}")

输出：

zee:{'bins': {'fix': [55, 0.1, '35e3']}, 'Title': 'é isso ai', 'Xlabel': 'como assim', 'Ylabel': 'Z #rightarrow e^{+}e^{-}', 'ymax': '35e3', 'ymim': 0.1, 'color': 4, 'scale': 1.0}

ztautau:{'ymax': '35e3', 'ymim': 0.1, 'ymaxp': 1.19, 'ymimp': 0.81, 'label': 'Z #rightarrow e^{+}e^{-}', 'color': ''}

ttbar:{'ymax': '35e3', 'ymim': 0.1, 'ymaxp': 1.19, 'ymimp': 0.81, 'label': 'Z #rightarrow e^{+}e^{-}'}

wplustaunu:{'bins': {'fix': [55, 0.1, '35e3']}, 'Title': 'é isso ai', 'Xlabel': 'como assim', 'Ylabel': 'Z #rightarrow e^{+}e^{-}', 'ymax': '35e3', 'ymim': 0.1, 'color': 'root.kAzure+3', 'scale': 1.0}

只有钥匙吗，先生？

[k for k,v in di.get("Histograms",{}).items() if isinstance(v, dict)]