私は次のコードを持っています
result = itertools.combinations_with_replacement(range(3),3)
for each in result:
print(each)
出力付き:
(0, 0, 0)
(0, 0, 1)
(0, 0, 2)
(0, 1, 1)
(0, 1, 2)
(0, 2, 2)
(1, 1, 1)
(1, 1, 2)
(1, 2, 2)
(2, 2, 2)
個々の「アイテム」を合計すると、たとえば2になる場合は、numpy配列として「result」に格納したいと思います。itertoolsが出力するデータ型が正確にわかりません。
擬似コードの例:
for each in result:
if sum(each)==2:
numpy array = each
したがって、理解を使用します。
import itertools
import numpy as np
result = itertools.combinations_with_replacement(range(3),3)
desired = [np.array(i) for i in result if sum(i)==2]
desired
#[array([0, 0, 2]), array([0, 1, 1])]
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加