def xmerge(a, b):
alen, blen = len(a), len(b)
mlen = min(alen, blen)
for i in xrange(mlen):
yield a[i]
yield b[i]
if alen > blen:
for i in xrange(mlen, alen):
yield a[i]
else:
for i in xrange(mlen, blen):
yield b[i]
a = [1, 2, 3]
b = [5, 6, 7, 8, 9, 10]
c = [i for i in xmerge(a, b)]
print c
c = [i for i in xmerge(b, a)]
print c
def roundrobin(*iterables):
"roundrobin('ABC', 'D', 'EF') --> A D E B F C"
# Recipe credited to George Sakkis
pending = len(iterables)
nexts = cycle(iter(it).next for it in iterables)
while pending:
try:
for next in nexts:
yield next()
except StopIteration:
pending -= 1
nexts = cycle(islice(nexts, pending))
def slove(a, b):
c = []
i = 0
j = 0
while i<len(a) and j<len(b):
c.append(a[i])
c.append(b[j])
i += 1
j += 1
while i < len(a):
c.append(a[i])
i += 1
while j < len(b):
c.append(b[j])
j += 1
修正: 之前的代码有问题, 更新一次..
不知道算不算优雅, 但应该省内存:
原来stackoverflow已经讨论过,写法都很吊哦。我个人很喜欢这个:
调用的cycle/islice函数都来自itertools
这样?
//写到一半搜了下python的三元表达式想起以前也是这种神兽飞过的心情用python的……
//不过还有三元写成
if a then b else c
的语言,现在还是能理解了……有个问题,如果这几个序列不都等长该怎么办。上面的解法是按长度最小的来:
但是如果不按最小的来该怎么办?补字符?补什么字符?
优雅的做数据处理,scipy系列库还是需要的。
有现成matplotlib中的flatten函数可以用。
a = [1, 2, 3]
b = [4, 5, 6]
一般的方法
def slove(a, b):
c = []
i = 0
j = 0
while i<len(a) and j<len(b):
c.append(a[i])
c.append(b[j])
i += 1
j += 1
while i < len(a):
c.append(a[i])
i += 1
while j < len(b):
c.append(b[j])
j += 1
if name == 'main':
slove(a, b)
如果只是列表的merge,是不是可以转成set(),再做交集操作
set作list merge时比list快。
但 @lohocla4dam 帮忙指出了不足