Python用于过滤另一个csv列表上以逗号分隔的列表

时间:2013-05-26 12:05:10

标签: python

我有两个字符串:

s1 = "Brendon, Melissa, Jason, , McGuirk" #the gauranteed string in format "x, y, z"
s2 = "brandon,melissa,jxz  ,paula,coach" #the messy string

并且想创建一个Python(2.7)列表,如果它存在则使用l1中的值,否则传递l2中的值。我有工作代码,但即使有列表推导,我觉得可能有更多Pythonic方式这样做。任何想法可能是什么?

l1 = [x.strip() for x in s1.split(',')]
l2 = [x.strip() for x in s2.split(',')]
f = lambda s: s[1] if s[1] else s[0]
final = [f(x) for x in zip(l2, l1)]

“final”列表现在包含:

['Brendon', 'Melissa', 'Jason', 'paula', 'McGuirk']

哪个是正确的。

-------编辑 所以,看看Jon的答案,a或b似乎是最简单,最易读的方法。我将字符串清理移动到一个小功能,最后得到了这个。有什么进一步的改进吗?

trim_csv = lambda csv: [s.strip() for s in csv.split(',')]
print [a or b for a, b in zip(trim_csv(s1), trim_csv(s2))]

2 个答案:

答案 0 :(得分:6)

适用于您的示例

s1 = "Brendon, Melissa, Jason, , McGuirk"
s2 = "brandon, melissa, jxz, paula, coach"

print [a or b for a, b in zip(s1.split(', '), s2.split(', '))]

稍微更通用的一个可以适应的:

import re
from itertools import izip_longest, ifilter, imap

s1 = "Brendon, Melissa, Jason, , McGuirk"
s2 = "brandon, melissa, jxz, paula, coach"


def take_first_not_empty(*args):
    splitter = re.compile(r'\s*?,\s*').split
    words = imap(splitter, args)
    return [next(ifilter(None, vals), '') for vals in izip_longest(*words, fillvalue='')]

答案 1 :(得分:2)

这样的东西?

>>> s1 = "Brendon, Melissa, Jason, , McGuirk"
>>> s2 = "brandon, melissa, jxz, paula, coach"
>>> [x if x else y  for x,y in zip( s1.split(', '),s2.split(', '))]
['Brendon', 'Melissa', 'Jason', 'paula', 'McGuirk']