python - 简化 Python 中多重哈希的使用

Question

我有一个大约 700 行 3 列的 CSV 文件，包含 label、rgb 和 string 信息，例如:

str;      rgb;                   label;         color
bones;    "['255','255','255']"; 2;             (241,214,145)
Aorta;    "['255','0','0']";     17;            (216,101,79)
VenaCava; "['0','0','255']";     16;            (0,151,206)

我想创建一种简单的方法来将一个唯一的输入转换为一个唯一的输出。

一种解决方案是将所有 ROIDisplayColor 条目与相应的标签条目作为字典进行哈希处理，例如rgb2label:

with open("c:\my_file.csv") as csv_file:
    rgb2label, label2rgb = {}, {} # rgb2str, label2str, str2label...
    for row in csv.reader(csv_file):
        rgb2label[row[1]] = row[2]
        label2rgb[row[2]] = row[1]

可以简单地使用如下:

>>> rgb2label[ "['255','255','255']"]
'2'
>>> label2rgb['2']
"['255','255','255']"

应用程序很简单，但每个关系都需要一个唯一的字典(rgb2label、rgb2str、str2rgb、str2label等...)。

是否存在具有相同易用性的更紧凑的解决方案？

Answer 1

在这里，您将自己限制在一对一的字典中，因此您最终会得到大量的字典(此处为 4^2=16)。

您可以使用一对多字典，这样您就只有 4 个:

for row in csv.reader(csv_file):
    rgb[row[1]] = row
    label[row[2]] = row

你会这样使用:

>>> rgb[ "['255','255','255']"][2]
'2'
>>> label['2'][1]
"['255','255','255']"

<小时/>

您也可以通过将行转换为 dict 来使这一点更清楚:

for row in csv.reader(csv_file):
    name, rgb, label, color = row
    d = {"rgb": rgb, "label": label}
    rgb[row[1]] = d
    label[row[2]] = d

你会这样使用:

>>> rgb[ "['255','255','255']"]["label"]
'2'
>>> label['2']["rgb"]
"['255','255','255']"