Python交错合并列表如何操作更优雅
比如a = [1, 2, 3], b = [4, 5, 6]
合并为[1, 4, 2, 5, 3, 6]
你觉得怎么写比较优雅?
a.extend(b)
python
# Python 3 def interpolate(*seqs): for items in zip(*seqs): yield from items >>> list(interpoltae('string', 'asdfgh')) ['s', 'a', 't', 's', 'r', 'd', 'i', 'f', 'n', 'g', 'g', 'h']
有个问题,如果这几个序列不都等长该怎么办。上面的解法是按长度最小的来:
python
>>> list(interpolate('string', 'asdf')) ['s', 'a', 't', 's', 'r', 'd', 'i', 'f']
但是如果不按最小的来该怎么办?补字符?补什么字符?
原来stackoverflow已经讨论过,写法都很吊哦。我个人很喜欢这个:
调用的cycle/islice函数都来自itertools
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))
python
def xmerge(a, b): tmp = (list(a), list(b)); return [tmp[i%2].pop(0) if tmp[i%2] else tmp[1-i%2].pop(0) for i in xrange(0, len(a) + len(b))] print xmerge([1,2,3], [5,6,7,8,9]) print xmerge([1,2,3,4,5], [7,8,9])
这样?
//写到一半搜了下python的三元表达式想起以前也是这种神兽飞过的心情用python的……
//不过还有三元写成if a then b else c
的语言,现在还是能理解了……
修正: 之前的代码有问题, 更新一次..
不知道算不算优雅, 但应该省内存:
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
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
print(c)
if name == 'main':
slove(a, b)
a = [1,2,3]
b = [4,5,6]
s = []
for i in zip(a,b):s.extend(list(i))
print s
result:
[1, 4, 2, 5, 3, 6]
python3.3
result = [list(zip(a, b))[i][j] for i in range(len(a)) for j in range(len(list(zip(a, b))[0]))]
虽然是一行,但是有点牵强,小括号太多
a=[1,2,3]
b=[4,5,6]
a=set(a)
b=set(b)
c=list(a|b)
如果只是python列表的merge,是不是可以转成set(),再做交集操作
set作list merge时比list快。
但 @lohocla4dam 帮忙指出了不足
python
from itertools import chain list(chain.from_iterable(zip(a, b))) # py2 list(chain(*zip(a, b)))
优雅的做数据处理,scipy系列库还是需要的。
有现成matplotlib中的flatten函数可以用。
from matplotlib.cbook import flatten
a = [1, 2, 3]
b = [4, 5, 6]
list(flatten(zip(a,b)))
玩蛇网文章,转载请注明出处和文章网址:https://www.iplaypy.com/wenda/wd18977.html
相关文章 Recommend
- • 2019年3月最新消息: Python 3.4.10 现已推出
- • [上海]招Python量化系统开发工程师
- • 优集品网络科技有限公司招Python中/高级工程师
- • 爱因互动科技发展有限公司招募Python开发攻城狮
- • mozio招聘Python/Django工程师
- • Kavout金融科技公司招Python研发工程师
- • Python数组逆向输出,编程练习题实例四十
- • Python数组插入排序,编程练习题实例三十九
- • Python矩阵for循环应用,编程练习题实例三十八
- • Python操作Redis数据库方面的问题
- • 请python高手帮我看看这段python代码中函数setter的
- • Python什么方法可以快速将两个队列变成字典
必知PYTHON教程 Must Know PYTHON Tutorials
- • python 解释器
- • python idle
- • python dir函数
- • python 数据类型
- • python type函数
- • python 字符串
- • python 整型数字
- • python 列表
- • python 元组
- • python 字典
- • python 集合
- • python 变量
- • python print
- • python 函数
- • python 类定义
- • python import
- • python help
- • python open
- • python 异常处理
- • python 注释
- • python continue
- • python pass
- • python return
- • python global
- • python assert
- • python if语句
- • python break
- • python for循环
- • python while循环
- • python else/elif
- • lambda匿名函数
必知PYTHON模块 Must Know PYTHON Modules
- • os 模块
- • sys 模块
- • re 正则表达式
- • time 日期时间
- • pickle 持久化
- • random 随机
- • csv 模块
- • logging 日志
- • socket网络通信
- • json模块
- • urlparse 解析URL
- • urllib 模块
- • urllib2 模块
- • robotparser 解析
- • Cookie 模块
- • smtplib 邮件
- • Base64 编码
- • xmlrpclib客户端
- • string 文本
- • Queue 线程安全
- • math数学计算
- • linecache缓存
- • threading多线程
- • sqlite3数据库
- • gzip压缩解压
最新内容 NEWS
- • django app提供pv信息的方法是什么
- • Django项目版本升级如何操作?
- • django较多数据传递如何优雅的呈现
- • django1.7获取参数问题求助
- • Django1.7使用内置comment遇到问题
- • python mysql数据库做insert操作时报_mysql_ex
- • 关于python mysql的duplicate insert机制的疑问
- • pymongo使用insert函数批量插入被中断要怎么
- • Python程序员解决棘手问题的常用库
- • 求助关于restfull api接口几个问题
图文精华 RECOMMEND
-
django1.7获取参数问题求助
-
Python程序员解决棘手问题的常用库
-
求问str()同__str__原理上有什么不同
-
scrapy框架里面用link extractor怎么能
-
python {}.fromkeys创建字典append添加操
-
python3 类型Type str doesn't support th
热点文章 HOT
- 学习Python有什么好的书籍推荐?
- Python匿名函数 Lambda表达式作用
- Python与Java、C、Ruby、PHP等编程语言有什么
- Python 正则中文网页字符串提取问题
- 如何为实时性应用存取经纬度?django my
- 想用python做个客户端,在二维码登录这个地
- 有让IDE可识别Python函数参数类型的方法吗
- Python字符串转换成列表正则疑问