将一个列表的元素多次映射到第二个列表的元素

Question

我有两个列表，我想将它们相互映射，但一个列表包含超过 1 个元素。所以我把它们压缩在一起，但它不能正常工作。

列表如下所示：

a = ['TEMP', 'TEMP,PRE', 'TEMP,HUM,RAN', 'HUM']
b = ['TEM', 'BAR', 'BAO', 'RAI']

我正在努力实现这一目标：

TEM.TEMP
BAR.TEMP
BAR.PRE
BAO.TEMP
BAO.HUM
BAO.RAN
RAI.HUM

我想将b 的每一项映射到a，但是在a 中有更多的值被, 分隔

我的代码如下：

import csv


mod1 = []
dev2 = []
d = {}
with open('/home/robi/Desktop/rob/device.csv', 'rb') as f:
    next(f, None)
    reader = csv.reader(f, delimiter=';')
    for row in reader:
        mod1.append(row[0])
        dev2.append(row[1])

    a = zip(dev2, mod1)
    for it, key in a:
        print it + '.' + key

但我得到这样的结果：

BAO.TEMP,HUM,RAN
BAR.TEMP,PRE
RAI.HUM

所以BAR 和BAO 没有正确映射。

Answer 1

zip 适用于list 索引，因此它无法根据您的标准识别特定项目中是否有更多项目。您必须进行一些后处理来构建最终列表，即用逗号将第一个列表中的元素拆分，然后将每个元素与第二个列表中的元素组合。比如：

a = ['TEMP', 'TEMP,PRE', 'TEMP,HUM,RAN', 'HUM']
b = ['TEM', 'BAR', 'BAO', 'RAI']

c = ["%s.%s"%(e[1],i) for e in zip(a, b) for i in e[0].split(",")]
# ['TEM.TEMP', 'BAR.TEMP', 'BAR.PRE', 'BAO.TEMP', 'BAO.HUM', 'BAO.RAN', 'RAI.HUM']

Answer 2

最后试试这个：

a = zip(dev2, mod1)
for it, key in a:
    words = key.split(',')
    for word in words:
        print it + '.' + word

Answer 3

使用list comprehension 和join（假设 a 和 b 的长度相同）

a = ['TEMP', 'TEMP,PRE', 'TEMP,HUM,RAN', 'HUM']
b = ['TEM', 'BAR', 'BAO', 'RAI']
length = len(a)

result = ['.'.join((b[index],sub_item) ) 
              for index in range(length) for sub_item in a[index].split(',')]

(这似乎比使用 zip 花费的时间少一点)

Answer 4

这是你要找的吗？

 >>> [[(b[i], x) for x in a[i].split(',')] for i in range(len(a))]
 [[('TEM', 'TEMP')],
 [('BAR', 'TEMP'), ('BAR', 'PRE')],
 [('BAO', 'TEMP'), ('BAO', 'HUM'), ('BAO', 'RAN')],
 [('RAI', 'HUM')]]

尽管我的回答对你来说不够好，但这里有一个解决方案，可以将它放入你想要的字符串中。

>>> reduce(lambda x, y: x + y, [['.'.join((b[i], x)) for x in a[i].split(',')] for i in range(len(a))])
['TEM.TEMP',
 'BAR.TEMP',
 'BAR.PRE',
 'BAO.TEMP',
 'BAO.HUM',
 'BAO.RAN',
 'RAI.HUM']

可悲的是，没有人会看到这个，但至少 我 会知道我有多棒...... /s

Answer 5

遍历b 中的项目，因为它在单个键中没有多个键，然后遍历a 中的项目并检查它是否包含分隔符, 并对其进行拆分。

如果需要，我还添加了代码以使用 set 进行唯一组合。

def getCombinations(a, b):
    combinations = []
    for bitem in b:
        for aitem in a:
            if ("," in aitem):
                for aitemInner in aitem.split(","):
                    combinations.append(bitem + "." + aitemInner)
            else:
                combinations.append(bitem + "." + aitem)
    ## Optional : if you want unique combinations of B and A
    unique = set(combinations)
    return unique

a = ['TEMP', 'TEMP,PRE', 'TEMP,HUM,RAN', 'HUM']
b = ['TEM', 'BAR', 'BAO', 'RAI']
combinations = getCombinations(a, b)
print("Keys in a          : " + str(len(a)))
print("Keys in b          : " + str(len(b)))
print("Total Combinations : " + str(len(combinations)))
print(combinations)

样品运行

Keys in a          : 4
Keys in b          : 4
Total Combinations : 16
{'TEM.HUM', 'RAI.HUM', 'BAR.PRE', 'BAO.HUM', 'TEM.PRE', 'RAI.RAN', 'RAI.TEMP', 'BAR.HUM', 'RAI.PRE', 'BAO.PRE', 'BAR.RAN', 'BAO.RAN', 'TEM.TEMP', 'TEM.RAN', 'BAO.TEMP', 'BAR.TEMP'}

EDIT ：更新解决方案，要求它不需要所有组合，但需要将 b 中的元素映射到 a。

在这里，如果您需要唯一的映射，我也使用set，否则您可以从get1To1Mapping() 方法返回组合。

NOTE：我得到了列表的最小大小，并且只给出了没有的映射。根据最小列表的元素，以避免异常。

def get1To1Mapping(a, b):
    combinations = []
    ## some might argue to iterate over len(b) but i would go for minimum
    ## of both the list to avoid exception
    for index in range(min(len(a), len(b))):
        if ("," in a[index]):
            for aitem in a[index].split(","):
                combinations.append(b[index] + "." + aitem)
        else:
            combinations.append(b[index] + "." + a[index])
    ## optional : if you want just unique mappings
    unique = set(combinations)
    return unique

a = ['TEMP', 'TEMP,PRE', 'TEMP,HUM,RAN', 'HUM']
b = ['TEM', 'BAR', 'BAO', 'RAI']
combinations = get1To1Mapping(a, b)
print("Keys in a          : " + str(len(a)))
print("Keys in b          : " + str(len(b)))
print("Total Combinations : " + str(len(combinations)))
print(combinations)

样品运行

Keys in a          : 4
Keys in b          : 4
Total Combinations : 7
{'BAR.PRE', 'RAI.HUM', 'TEM.TEMP', 'BAO.TEMP', 'BAO.HUM', 'BAO.RAN', 'BAR.TEMP'}

Answer 6

其他答案忽略了您正在阅读 csv 文件。您可以在阅读文件后立即以更合适的结构排列数据。将您的此类列表压缩在一起表明数据可以以更好的方式结构化 - 字典。

您可以避免创建两个列表，而是立即将元素添加到字典中（如果必须对元素进行排序，请使用 collections.OrderedDict）：

dictio = {}

with open('csv_file.csv') as f:
    next(f, None)
    reader = csv.reader(f, delimiter=';')
    for mod1, dev2 in reader:
        dictio[dev2] = mod1.split(',')

for key, value in dictio.items():
    for word in value:
        print('{}.{}'.format(key, word))