有效地将迭代器输出复制到numpy数组的一个片段

2024-09-28 05:20:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我想将itertools组合的结果复制到现有numpy数组的切片中。 下面的代码可以工作,但我想看看是否有更有效的方法来实现这一点,因为我将在模拟中重复数百万次

from itertools import combinations
import numpy as np


A=[2,5,7,3,100,8]
B=np.zeros((20,20),dtype=np.uint8)

numsims = 1

for i in range (numsims):
    combs = list(combinations(A,3))
    B[0:3,0:20]=np.array(combs).T

print (B)

Tags: 方法代码fromimportnumpyasnpzeros

热门问题