トップ Pythonに関する質問 Python　配列から要素をランダムに抽出し、クラス分けしたときの制約条件の作り方　組み合わせNG

編集履歴

回答編集履歴

ユーザ追加処理コード追加

2020/01/08 07:12

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -121,3 +121,107 @@
     return False
 ```
+ユーザ追加処理
+```python
+import random
+# ユーザーの情報の読み込み
+users = [
+  (1,0,1,0),
+  (2,0,0,0),
+  (3,0,1,0),
+  (4,0,0,0),
+  (5,1,0,0),
+  (6,1,0,0),
+  (7,0,0,0),
+  (8,1,0,0),
+  (9,0,0,0),
+  (10,0,1,0),
+  (11,0,0,12),
+  (12,0,0,13),
+  (13,0,0,0),
+  (14,0,0,0),
+  (15,0,1,0),
+  (16,0,0,3),
+  (17,0,0,0),
+  (18,0,0,0),
+  (19,0,0,9),
+]
+no20 = (20, 0, 1, 0)
+users.append(no20)
+# ユーザーをシャッフルにし、3クラスに分ける
+random.shuffle(users)
+c1 = len(users) * 1 // 3
+c2 = len(users) * 2 // 3
+classes = class1, class2, class3 = users[:c1], users[c1:c2], users[c2:]
+# クラス毎に、制約条件1が1人以下かつ、制約条件2が2人以下ならOK
+if all(sum(u[1] for u in c) <= 1 and sum(u[2] for u in c) <= 2 for c in classes):
+    print('OK')
+    # 組み合わせNGのメンバーがクラスにいなければOK
+    if all(all(u[3] != m[0] for m in c) for c in classes for u in c if u[3]):
+        print('組み合わせもOK')
+    else:
+        print('組み合わせはNG')
+else:
+    print('NG')
+print(class1)
+print(class2)
+print(class3)
+```

コメントで質問された処理の展開を追記

2020/01/08 07:12

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -95,3 +95,29 @@
 print(class3)
 ```
+コメントで質問された処理を展開：
+```python
+def conflict(classes):
+    for class_ in classes:
+        for user in class_:
+            if user[3] != 0:
+                for member in class_:
+                    if user[3] == member[0]
+                        return True
+    return False
+```

コメント修正

2020/01/07 07:47

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -54,7 +54,7 @@
-# ユーザーをシャッフルにし、3組に分ける
+# ユーザーをシャッフルにし、3クラスに分ける
 random.shuffle(users)

ユーザ情報をタプルに変更

2020/01/07 04:27

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -12,43 +12,43 @@
 users = [
-  [1,0,1,0],
+  (1,0,1,0),
-  [2,0,0,0],
+  (2,0,0,0),
-  [3,0,1,0],
+  (3,0,1,0),
-  [4,0,0,0],
+  (4,0,0,0),
-  [5,1,0,0],
+  (5,1,0,0),
-  [6,1,0,0],
+  (6,1,0,0),
-  [7,0,0,0],
+  (7,0,0,0),
-  [8,1,0,0],
+  (8,1,0,0),
-  [9,0,0,0],
+  (9,0,0,0),
-  [10,0,1,0],
+  (10,0,1,0),
-  [11,0,0,12],
+  (11,0,0,12),
-  [12,0,0,13],
+  (12,0,0,13),
-  [13,0,0,0],
+  (13,0,0,0),
-  [14,0,0,0],
+  (14,0,0,0),
-  [15,0,1,0],
+  (15,0,1,0),
-  [16,0,0,3],
+  (16,0,0,3),
-  [17,0,0,0],
+  (17,0,0,0),
-  [18,0,0,0],
+  (18,0,0,0),
-  [19,0,0,9],
+  (19,0,0,9),
 ]

c1,c2計算方法変更

2020/01/07 04:26

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -58,9 +58,9 @@
 random.shuffle(users)
-c1 = len(users) // 3
+c1 = len(users) * 1 // 3
-c2 = c1 * 2
+c2 = len(users) * 2 // 3
 classes = class1, class2, class3 = users[:c1], users[c1:c2], users[c2:]

コメント修正

2020/01/07 04:21

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -54,7 +54,7 @@
-# ユーザーの中から、1組に入れる7人をランダムに選ぶ
+# ユーザーをシャッフルにし、3組に分ける
 random.shuffle(users)
@@ -66,15 +66,17 @@
-# 各組に入っている人の内訳が、制約条件1が1人以下かつ、制約条件2が2人以下のとき、
+# クラス毎に、制約条件1が1人以下かつ、制約条件2が2人以下ならOK
 if all(sum(u[1] for u in c) <= 1 and sum(u[2] for u in c) <= 2 for c in classes):
     print('OK')
-    if all(all(a[3] != b[0] for b in c) for c in classes for a in c if a[3]):
+    # 組み合わせNGのメンバーがクラスにいなければOK
+    if all(all(u[3] != m[0] for m in c) for c in classes for u in c if u[3]):
-        print('組み合わせはOK')
+        print('組み合わせもOK')
     else:

ユーザ情報の先頭要素を1からの出席番号に変更、最後の要素も出席番号

2020/01/07 04:16

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -12,27 +12,25 @@
 users = [
-  [0,0,1,0],
+  [1,0,1,0],
-  [1,0,0,0],
+  [2,0,0,0],
-  [2,0,1,0],
+  [3,0,1,0],
-  [3,0,0,0],
+  [4,0,0,0],
-  [4,1,0,0],
   [5,1,0,0],
-  [6,0,0,0],
+  [6,1,0,0],
-  [7,1,0,0],
+  [7,0,0,0],
-  [8,0,0,0],
+  [8,1,0,0],
-  [9,0,1,0],
+  [9,0,0,0],
-  [10,0,0,0],
+  [10,0,1,0],
   [11,0,0,12],
@@ -40,15 +38,17 @@
   [13,0,0,0],
-  [14,0,1,0],
+  [14,0,0,0],
-  [15,0,0,3],
+  [15,0,1,0],
-  [16,0,0,0],
+  [16,0,0,3],
   [17,0,0,0],
-  [18,0,0,9],
+  [18,0,0,0],
+  [19,0,0,9],
 ]

組み合わせ対応

2020/01/07 04:08

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -1,6 +1,4 @@
 こんな書き方はいかが？
-組み合わせNGが抜けてたのでのちほど修正します。
@@ -36,9 +34,9 @@
   [10,0,0,0],
-  [11,0,0,11],
+  [11,0,0,12],
-  [12,0,0,11],
+  [12,0,0,13],
   [13,0,0,0],
@@ -74,6 +72,14 @@
     print('OK')
+    if all(all(a[3] != b[0] for b in c) for c in classes for a in c if a[3]):
+        print('組み合わせはOK')
+    else:
+        print('組み合わせはNG')
 else:
     print('NG')

修正中

2020/01/07 03:40

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -1,4 +1,6 @@
 こんな書き方はいかが？
+組み合わせNGが抜けてたのでのちほど修正します。

バグ修正

2020/01/07 03:30

投稿

shiracamus

スコア5406

test CHANGED Viewed

@@ -68,7 +68,7 @@
 # 各組に入っている人の内訳が、制約条件1が1人以下かつ、制約条件2が2人以下のとき、
-if all(sum(u[1] for u in c) <= 1 or sum(u[2] for u in c) <= 2 for c in classes):
+if all(sum(u[1] for u in c) <= 1 and sum(u[2] for u in c) <= 2 for c in classes):
     print('OK')