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

編集履歴

回答編集履歴

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

2020/01/08 07:12

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -59,4 +59,56 @@
                     if user[3] == member[0]
                         return True
     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

answer CHANGED Viewed

@@ -46,4 +46,17 @@
 print(class1)
 print(class2)
 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

answer CHANGED Viewed

@@ -26,7 +26,7 @@
   (19,0,0,9),
 ]
-# ユーザーをシャッフルにし、3組に分ける
+# ユーザーをシャッフルにし、3クラスに分ける
 random.shuffle(users)
 c1 = len(users) * 1 // 3
 c2 = len(users) * 2 // 3

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

2020/01/07 04:27

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -5,25 +5,25 @@
 # ユーザーの情報の読み込み
 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),
 ]
 # ユーザーをシャッフルにし、3組に分ける

c1,c2計算方法変更

2020/01/07 04:26

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -28,8 +28,8 @@
 # ユーザーをシャッフルにし、3組に分ける
 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:]
 # クラス毎に、制約条件1が1人以下かつ、制約条件2が2人以下ならOK

コメント修正

2020/01/07 04:21

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -26,17 +26,18 @@
   [19,0,0,9],
 ]
-# ユーザーの中から、1組に入れる7人をランダムに選ぶ
+# ユーザーをシャッフルにし、3組に分ける
 random.shuffle(users)
 c1 = len(users) // 3
 c2 = c1 * 2
 classes = class1, class2, class3 = users[:c1], users[c1:c2], users[c2:]
-# 各組に入っている人の内訳が、制約条件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')
+    # 組み合わせNGのメンバーがクラスにいなければOK
-    if all(all(a[3] != b[0] for b in c) for c in classes for a in c if a[3]):
+    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:
         print('組み合わせはNG')
 else:

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

2020/01/07 04:16

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -5,25 +5,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],
   [12,0,0,13],
   [13,0,0,0],
-  [14,0,1,0],
+  [14,0,0,0],
+  [15,0,1,0],
-  [15,0,0,3],
+  [16,0,0,3],
-  [16,0,0,0],
   [17,0,0,0],
-  [18,0,0,9],
+  [18,0,0,0],
+  [19,0,0,9],
 ]
 # ユーザーの中から、1組に入れる7人をランダムに選ぶ

組み合わせ対応

2020/01/07 04:08

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -1,5 +1,4 @@
 こんな書き方はいかが？
-組み合わせNGが抜けてたのでのちほど修正します。
 ```python
 import random
@@ -17,8 +16,8 @@
   [8,0,0,0],
   [9,0,1,0],
   [10,0,0,0],
-  [11,0,0,11],
+  [11,0,0,12],
-  [12,0,0,11],
+  [12,0,0,13],
   [13,0,0,0],
   [14,0,1,0],
   [15,0,0,3],
@@ -36,6 +35,10 @@
 # 各組に入っている人の内訳が、制約条件1が1人以下かつ、制約条件2が2人以下のとき、
 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]):
+        print('組み合わせはOK')
+    else:
+        print('組み合わせはNG')
 else:
     print('NG')

修正中

2020/01/07 03:40

投稿

shiracamus

スコア5406

answer CHANGED Viewed

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

バグ修正

2020/01/07 03:30

投稿

shiracamus

スコア5406

answer CHANGED Viewed

@@ -33,7 +33,7 @@
 classes = class1, class2, class3 = users[:c1], users[c1:c2], users[c2:]
 # 各組に入っている人の内訳が、制約条件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')
 else:
     print('NG')