76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
|
|
import json
|
|
data = json.load(open('boardgames-2022.json', 'r'))
|
|
data1 = list(filter(lambda v: v['plays'] == 1, data))
|
|
data2 = list(filter(lambda v: v['plays'] == 2, data))
|
|
data3 = list(filter(lambda v: v['plays'] == 3, data))
|
|
|
|
def plotx(xlambda, games, color, marker, plt):
|
|
x = list(map(xlambda, games))
|
|
y = list(map(lambda v: v['weight'], games))
|
|
|
|
plt.grid(linestyle='dotted')
|
|
|
|
plt.scatter(x, y, s=150, c=color, marker=marker, alpha=0.5)
|
|
plt.set_ylim(0, 4.5)
|
|
|
|
|
|
plt.set_xlim = plt.xlim
|
|
plt.set_ylim = plt.ylim
|
|
|
|
fig, ((ax1, ax2)) = plt.subplots(2, figsize=(7, 7))
|
|
|
|
ax1.set_xlim(1, 10)
|
|
ax1.set_xlabel('BGG Score (1-10)')
|
|
ax1.set_ylabel('Weight (1-5)')
|
|
plotx(lambda v: v['score']['BGG'], data1, 'red', 'v', ax1)
|
|
plotx(lambda v: v['score']['BGG'], data2, 'orange', ',', ax1)
|
|
plotx(lambda v: v['score']['BGG'], data3, 'green', '^', ax1)
|
|
ax2.set_xlim(0, 5)
|
|
ax2.set_xlabel('Own Score (1-4)')
|
|
ax2.set_ylabel('Weight (1-5)')
|
|
plotx(lambda v: v['score']['W'], data1, 'red', 'v', ax2)
|
|
plotx(lambda v: v['score']['W'], data2, 'orange', ',', ax2)
|
|
plotx(lambda v: v['score']['W'], data3, 'green', '^', ax2)
|
|
|
|
print("Correlations between BGG score and Own score")
|
|
print(" \t -- PLAY 1")
|
|
print(np.corrcoef(list(map(lambda v: v['score']['BGG'], data1)), list(map(lambda v: v['score']['W'], data1))))
|
|
print(" \t -- PLAY 2")
|
|
print(np.corrcoef(list(map(lambda v: v['score']['BGG'], data2)), list(map(lambda v: v['score']['W'], data2))))
|
|
print(" \t -- PLAY 3")
|
|
print(np.corrcoef(list(map(lambda v: v['score']['BGG'], data3)), list(map(lambda v: v['score']['W'], data3))))
|
|
print(" \t -- OVERALL")
|
|
print(np.corrcoef(list(map(lambda v: v['score']['BGG'], data)), list(map(lambda v: v['score']['W'], data))))
|
|
print("\n\n")
|
|
|
|
#print("Correlation between weight and plays?")
|
|
#print(np.corrcoef(list(map(lambda v: v['score']['W'], data)), list(map(lambda v: v['weight'], data))))
|
|
|
|
|
|
def play(p):
|
|
if p == 1:
|
|
return "-"
|
|
if p == 2:
|
|
return "+"
|
|
return "++"
|
|
|
|
for e in data:
|
|
print("| [" + e['name'] + "](" + e['link'] + ") | " + str(e['score']['BGG']) + " (" + str(e['score']['W']) + ") | " + str(e['weight']) + " | " + play(e['plays']) + " |")
|
|
|
|
print("Averages:")
|
|
print("\t BGG")
|
|
print(str(sum(list(map(lambda v: v['score']['BGG'], data))) / len(data)))
|
|
print("\t Own")
|
|
print(str(sum(list(map(lambda v: v['score']['W'], data))) / len(data)))
|
|
print("\t Weight")
|
|
print(str(sum(list(map(lambda v: v['weight'], data))) / len(data)))
|
|
print("\t plays")
|
|
print(str(sum(list(map(lambda v: v['plays'], data))) / len(data)))
|
|
|
|
|
|
plt.show()
|
|
|