Apriori-Algorithm/generateDatabse.py at master · MrPatel95/Apriori-Algorithm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import random

computerStuff = ["Desktop", "Laptop", "Mouse", "Keyboard", "Printer", "HDMI", "SSD", "HDD", "Monitor", "Processor"]
candies = ["AlmondJoy", "KitKat", "HeathBar", "BabyRuth", "Twix", "Snickers", "Cotton Candy", "MilkyWay", "Hershey", "Cadbury"]
mobileStuff = ["HeadPhones", "Charger", "Splitter", "Lences", "Battery", "Processor", "HomeButton", "Speakers", "Camera", "Screen"]
food = ["Bread", "Milk", "Eggs", "Tea", "Coffee", "Cream", "Bagel", "Sugar", "CreamCheese", "Cheese"]
automobile = ["SideLight", "HeadLight", "Engine", "SteeringWheel", "WindShield", "Oil", "BreakPad", "Wheels", "Tires", "Car"]

selected = []

#File 1
file = open('computerStuff.txt' ,'a')
file.truncate()
for j in range(25):
    for x in range(10):
        selected.append(computerStuff[random.randint(0,9)])
    selected = set(selected)
    selected = list(selected)
    for i in range(len(selected)):
        if i != len(selected) - 1:
            file.write(selected[i] + ',')
        else:
            file.write(selected[i])
    selected = []
    file.write('\n')

#File 2
file = open('candies.txt' ,'a')
file.truncate()
for j in range(25):
    for x in range(10):
        selected.append(candies[random.randint(0,9)])
    selected = set(selected)
    selected = list(selected)
    for i in range(len(selected)):
        if i != len(selected) - 1:
            file.write(selected[i] + ',')
        else:
            file.write(selected[i])
    selected = []
    file.write('\n')

#File 3
file = open('mobileStuff.txt' ,'a')
file.truncate()
for j in range(25):
    for x in range(10):
        selected.append(mobileStuff[random.randint(0,9)])
    selected = set(selected)
    selected = list(selected)
    for i in range(len(selected)):
        if i != len(selected) - 1:
            file.write(selected[i] + ',')
        else:
            file.write(selected[i])
    selected = []
    file.write('\n')

#File 4
file = open('food.txt' ,'a')
file.truncate()
for j in range(25):
    for x in range(10):
        selected.append(food[random.randint(0,9)])
    selected = set(selected)
    selected = list(selected)
    for i in range(len(selected)):
        if i != len(selected) - 1:
            file.write(selected[i] + ',')
        else:
            file.write(selected[i])
    selected = []
    file.write('\n')

#File 5
file = open('automobile.txt' ,'a')
file.truncate()
for j in range(25):
    for x in range(10):
        selected.append(automobile[random.randint(0,9)])
    selected = set(selected)
    selected = list(selected)
    for i in range(len(selected)):
        if i != len(selected) - 1:
            file.write(selected[i] + ',')
        else:
            file.write(selected[i])
    selected = []
    file.write('\n')