python - 从 Python 3 中的可迭代对象生成值元组
问题描述
如果一个函数被传递了一个未定义数量的任意长度 M 的迭代器 N,有没有办法获得一个生成器,该生成器产生 M 个大小为 N 的元组,其中包含传递的迭代器的元素?
def dispatch(*iterables):
args = ( some kind of generator that involves *iterables)
for _ in args:
yield _
换句话说,如果我们将 *iterables 视为 NxM 矩阵,其中每一列(函数参数)都是可迭代的,是否有办法使用生成器来产生矩阵的行?
例如:
a = [9,8,7,6]
b = 'ciao'
c = iter(range(0,4))
>>> res = dispatch(a,b,c)
>>> res.__next__()
(9,c,0)
>>> res.__next__()
(8,i,1)
etc...
此外,由于这个函数也可以只接受 1 个可迭代作为参数,所以生成器应该能够处理这种情况并输出如下内容:
a = [9,8,7,6]
>>> res = dispatch(a)
>>> res.__next__()
(9,)
>>> res.__next__()
(8,)
我尝试使用 zip 但它不能处理上述边缘情况,而且它有点慢,这表明它可能必须在产生输出之前阅读整个内容。
解决方案
您可以map
与包装参数结合使用lambda
:
>>> list(map(lambda *x: tuple(x), range(10), range(10), range(10)))
[(0, 0, 0), (1, 1, 1), (2, 2, 2), (3, 3, 3), (4, 4, 4), (5, 5, 5), (6, 6, 6), (7, 7, 7), (8, 8, 8), (9, 9, 9)]
>>> list(map(lambda *x: tuple(x), range(10)))
[(0,), (1,), (2,), (3,), (4,), (5,), (6,), (7,), (8,), (9,)]
您的功能将非常简单:
def dispatch(*args):
return map(lambda *x: tuple(x), *args)
如您的示例所示:
>>> a = [9,8,7,6]
>>> b = 'ciao'
>>> c = iter(range(0,4))
>>> list(dispatch(a, b, c))
[(9, 'c', 0), (8, 'i', 1), (7, 'a', 2), (6, 'o', 3)]
推荐阅读
- node.js - Vue|Nuxt - 将项目复制到生产服务器导致错误
- reactjs - 如何将边框底部放在reactjs中的react-chips?
- c# - 如何获得 Hyper-V VM 来宾操作系统
- ruby-on-rails - 如何访问哈希中的嵌套键
- julia - 使用 Julia 无法在 Atom 中打开调试器窗格
- arm - 带有 std::pair 的 API
在 ARM 上从 C++14 切换到 C++17 时中断? - json - 如何正确地将新密钥传递给 JSON 字典?
- matlab - 将面顶点网格导出到 STL
- javascript - 我想在我的代码中多次调用一个函数?
- javascript - 使用 Passport.js 和 Sequelize 引用多个模型