如何改进这个List Comprehensions?

我想知道我可以改进我的python代码,使其写得更短。我只使用标准库和numpy库。下面是代码。

import numpy as np
x=1
y=2
np.array([(a,b,c) for a in [x,y] for b in [x,y] for c in [x,y]])

正如你所看到的,这个程序产生了所有三个元素的组合,在这个程序中有一个类似的部分,如 for a in [x, y] 并且可以写得更短吗?

[编辑]

我想只使用标准库和numpy库。

输入应该是任何值,比如x=3,y=8。

解决方案:

使用 itertools.product:

from itertools import product

# ...
np.array(list(product((x, y), repeat=3)))

给TA打赏
共{{data.count}}人
人已打赏
未分类

具有Fusionauth passport-oauth OAuth2Strategy错误的节点。获取访问令牌失败

2022-9-9 9:43:20

未分类

非const lvalue的'const int *'类型的引用不能绑定到一个无关的'int *'类型的值。

2022-9-9 9:54:16

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索