21 lines
555 B
Python
21 lines
555 B
Python
import numpy as np
|
|
from pathlib import Path
|
|
|
|
filepath = Path("./data/1_example")
|
|
filepath = Path("./data/1_input")
|
|
|
|
with open(filepath, 'r') as filein:
|
|
data = np.transpose([[int(i) for i in l.split()] for l in filein.readlines()])
|
|
data.sort()
|
|
data_diff = np.abs(np.diff(data,axis=0).flatten())
|
|
data_sum = data_diff.sum()
|
|
print(data_sum)
|
|
|
|
|
|
def score_similarity(a: np.ndarray, b: np.ndarray):
|
|
left = a# p.unique(a)
|
|
right = np.array([np.sum(np.where(b==i, 1, 0)) for i in left])
|
|
return np.sum(left*right)
|
|
|
|
|
|
print(score_similarity(*data))
|